from sys import path
from tqdm import tqdm
from PIL import Image, ImageDraw, ImageFont
def readline_drawrect(line,replaced):
list_info = []
for i in range(len(line.split(' '))):
list_info.append(line.split(' ')[i])
img= list_info[0]
path = img.replace("JPEGImages",replaced)
image = Image.open(img)
image = image.convert('RGB')
draw = ImageDraw.Draw(image)
for i in range(len(list_info)-1):
left = int(list_info[i+1].split(",")[0])
top = int(list_info[i+1].split(",")[1])
right = int(list_info[i+1].split(",")[2])
bottom = int(list_info[i+1].split(",")[3])
draw.rectangle(((left, top), (right, bottom)),fill=None,outline=(0,255,127), width=2)
del draw
image.save(path)
return image,path
if __name__=="__main__":
with open("2007_test.txt", "r", encoding="utf-8") as f:
text = f.readlines()
for i in tqdm(range(len(text)),desc='进行中',ncols=100):
aoligei = text[i]
aoligei = aoligei.replace("VOCdevkit_","VOCdevkit")
image,path = readline_drawrect(aoligei,replaced="annotated")