#!/usr/bin/env python
# -*- coding:utf-8 -*-

# 本代码主要用来根据.txt文档中的坐标绘制anchor
# 需要基于yolo.py中的detec部分代码绘制相关的rectangle

from sys import path
from tqdm import tqdm
from PIL import Image, ImageDraw, ImageFont

def readline_drawrect(line,replaced):
    # 读取每一行内容, 拆解成[0]: 路径, [1,2,3]:坐标以及标签, 由于目前应用的是单类别的检测, 所以没有class的相关内容
    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__":
    # txt文档中每一行的信息
    # information = '../VOCdevkit/VOC2007/JPEGImages/xxxxx.jpg 202,203,216,215,0 129,43,144,72,0'


    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")