import cv2 as cv
font = cv.FONT_HERSHEY_SIMPLEX
def get_file_name(dir_path):
import os
name = []
if os.path.isdir(dir_path): #isdir确定文件夹
for filename in os.listdir(dir_path): # 获取当前路径下的文件名
name.append( filename)
name.sort() #升序排序
return name
FILE_NAMES = get_file_name(r"/home/wanys/data/mydata/ssd/data/training/label")
for names_image in FILE_NAMES:
f=open("/home/wanys/data/mydata/ssd/data/training/label/"+names_image)
new_name=names_image.split(".")
new_names=new_name[0]+".png"
img=cv.imread("/home/wanys/data/mydata/ssd/data/training/image/"+new_names)
for ff in f.readlines():
fff=ff.split(" ")
sign=str(fff[0])
sign_x=fff[4]
sign_y=fff[5]
sign_w=fff[6]
sign_h=fff[7]
sign_x=sign_x.split(".")
a=int(sign_x[0])
sign_y=sign_y.split(".")
b=int(sign_y[0])
sign_w=sign_w.split(".")
c=int(sign_w[0])
sign_h=sign_h.split(".")
d=int(sign_h[0])
if sign=="warning":
cv.putText(img, sign, (a,b), cv.FONT_HERSHEY_PLAIN, 1.5, (30, 100, 100), 2)
elif sign=="prohibitory":
cv.putText(img, sign, (a,b), cv.FONT_HERSHEY_PLAIN, 1.5, (0, 0, 255), 2)
elif sign=="mandatory":
cv.putText(img, sign, (a,b), cv.FONT_HERSHEY_PLAIN, 1.5, (255, 255, 0), 2)
cv.rectangle(img, (a, b), (c, d), (0, 255, 0), 2)
cv.imwrite("/home/wanys/data/mydata/ssd/data/training/out/"+new_names,img)
检查kitti数据集标注正确性
最新推荐文章于 2023-08-30 23:31:52 发布