将txt中的标注画到图片数据集中

import cv2
import os
import os.path

def select(text_path, outpath, image_path):
    text_file = open(text_path, 'r')
    line = text_file.readline()
    text_tables = []
    i = 0
    while line:
        text_data = eval(line)
        text_tables.append(text_data)
        line = text_file.readline()
    for filename in os.listdir(image_path):
        img = cv2.imread(image_path + '/' + filename)
        j = len(text_tables)
        if i < j+1:
            x1 = int(text_tables[i][0])
            y1 = int(text_tables[i][1])
            w = int(text_tables[i][2])
            h= int(text_tables[i][3])
            x2 = x1+w
            y2 = y1+h
            cv2.rectangle(img, (x1, y1), (x2, y2), (255, 0, 0), thickness=2)
            cv2.putText(img, 'gabion', (x1, y1 - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.75, (0, 0, 255), thickness=2)
            i = i + 1
        cv2.imwrite(os.path.join(outpath, filename + '.jpg'), img)

if __name__ == '__main__':
    image_path = r'F:/ljwdata/data/UAV123/data_seq/UAV123/bike3/'
    txt_path = r'F:\ljwdata\data\UAV123\anno\UAV123\bike3.txt'
    outpath = r'F:\ljwdata\data\test'
    select(txt_path, outpath, image_path)