import cv2
import os
img_dir = 'D:/images/'
label_dir = 'D:/labels/'
save_dir = 'D:/save/' # 事先新建一个文件夹,用来存放标注好的图片
lable_file = os.listdir(label_dir)
img_file = os.listdir(img_dir)
for file in lable_file:
file_dir = os.path.join(label_dir, file)
with open(file_dir, 'r') as f:
print(os.path.join(img_dir, file.split('.')[0]+'.jpg'))
image_src = cv2.imread(os.path.join(img_dir, file.split('.')[0]+'.jpg'))
image_row = image_src.shape[0]
image_col = image_src.shape[1]
for line in f.readlines():
x_ = float(line.split(' ')[1])
y_ = float(line.split(' ')[2])
w_ = float(line.split(' ')[3])
h_ = float(line.split(' ')[4])
w = image_col
h = image_row
x1 = w * x_ - 0.5 * w * w_
x2 = w * x_ + 0.5 * w * w_
y1 = h * y_ - 0.5 * h * h_
y2 = h * y_ + 0.5 * h * h_
draw = cv2.rectangle(image_src,(int(x1),int(y1)),(int(x2),int(y2)),[0,255,0],2)
cv2.imwrite(os.path.join(save_dir, file.split('.')[0]+'.jpg'), draw)
批量可视化目标检测标注框——YOLO格式数据集
最新推荐文章于 2024-07-02 21:00:49 发布