检查kitti数据集标注正确性

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)
 
  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

和她一起热爱生活

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值