根据yolo标签批量画框

最新推荐文章于 2024-08-30 17:32:45 发布

追梦小狂魔

最新推荐文章于 2024-08-30 17:32:45 发布

阅读量187

点赞数

文章标签： YOLO python opencv

本文链接：https://blog.csdn.net/qq_15557299/article/details/132050421

版权

import glob
from tqdm import tqdm
import os
import shutil
import random
import cv2
txts=glob.glob("labels/*.txt")
for txt_path in tqdm(txts):
    img_path=txt_path.replace("labels","images").replace(".txt",".jpg")
    with open(txt_path)as f:
        data=f.readlines()
    image = cv2.imread(img_path)

    # 调整图像大小
    resized_image = cv2.resize(image, (640, 480))
    for i in data:
        id,xmin,ymin,xmax,ymax=i.split(" ")
        # print(id,xmin,ymin,xmax,ymax)
        cv2.rectangle(resized_image, (int(float(xmin) * 640), int(float(ymin) * 480)), (int(float(xmax) * 640), int(float(ymax) * 480)),
                      (255, 0, 0), 1)
    # print(resized_image.shape)
    img_save=os.path.join("biaoji", os.path.basename(img_path))
    # print(img_save)
    cv2.imwrite(img_save,resized_image)