检查JPG与其标签的XML是否一一对应

检查JPG与其标签的XML是否一一对应

在手动标注自己的数据集,或请人帮忙标注数据集之后,应对检查是否每一个JPG文件都有一个对应的XML文件(检查是否标全),每一个XML文件是否都对应一个JPG文件(图片标了觉得图片太模糊而将图片删除,会遗留多余的XML文件)
修改JpgDir为JPG所在目录,XmlDir为XML文件所在目录,具体代码如下:

import os

XmlDir = r"D:\Program Files (x86)\Pycharm\PycharmProject\pythonProject\yolo3-pytorch-master\VOCdevkit\VOC2007\Annotations"
JpgDir = r"D:\Program Files (x86)\Pycharm\PycharmProject\pythonProject\yolo3-pytorch-master\VOCdevkit\VOC2007\JPEGImages"

NoXml = []
NoJpg = []

for root, dirs, files in os.walk(JpgDir):
    for file in files:
        if file[-1] == 'g':
            if os.path.exists(XmlDir + "\\"+file[:-3] + "xml") is False:
                NoXml.append(XmlDir+"\\"+file)

for root, dirs, files in os.walk(XmlDir):
    for file in files:
        if file[-1] == 'l':
            if os.path.exists(JpgDir + "\\"+file[:-3] + "jpg") is False:
                NoJpg.append(JpgDir+"\\"+file)
if len(NoXml) == 0:
    print("All jpg are labeled")
else:
    print("%d unlabeled" % len(NoXml))
    print(NoXml)

if len(NoJpg) == 0:
    print("All xml have a jpg")
else:
    print("%d xmls have no jpg" % len(NoJpg))
    print(NoJpg)

运行结果如下:
在这里插入图片描述

  • 1
    点赞
  • 14
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值