Python批量检查labelimg数字标签是否标注错误

最新推荐文章于 2024-07-23 11:22:11 发布

你不过是人

最新推荐文章于 2024-07-23 11:22:11 发布

阅读量1.2k

点赞数 1

分类专栏：其它文章标签： labelimg 数字识别

本文链接：https://blog.csdn.net/qq_41833285/article/details/102895369

版权

在大规模标注数字图像数据集时，可能存在错误。本文介绍如何利用Python检查labelimg标注的数字标签，通过比较文件名后缀与xml文件中的标签，找出错误并修复，以确保数据准确性。

摘要由CSDN通过智能技术生成

在我们标注一个几千张的数据集的时候，难免会出错。而这些错误会被当成负样本，对我们的准确度造成影响。

以我为例，做仪表识别时，标注数千张数字图片。

大概是这样的。
示例图片
N002145-003 我们人工读数给文件名加后缀003（这一步没有捷径）

我们的核心思想用003和标注生成的xml文件中标签对比
在这里插入图片描述
那么就很容易实现了，这里使用python

import xml.etree.cElementTree as et
import os
path = r'D:/w h/test'
filenames = os.listdir(path)
for filename in filenames:
    tree=et.parse(filename)
    root=tree.getroot()
    filename=root.find('filename').text
    i = 7
    for Object in root.findall('object'):
        name=Object.find('name').text
        list1=[name]
        list5=filename
        i = i + 1
        if str(name)!=str(list5[i]):
            print("error")
            print(filename)

大功告成！（短短十几行代码）

在这里插入图片描述
居然有十八张错误0.0