如何判断图像标注数据的正误

(1)将annotations放到一个文件夹下,并读取标签。


 
 
  1. import os
  2. import glob
  3. import pandas as pd
  4. import xml.etree.ElementTree as ET
  5. def xml_to_csv(path):
  6. xml_list = []
  7. for xml_file in glob.glob(path + '/*.xml'):
  8. tree = ET.parse(xml_file)
  9. root = tree.getroot()
  10. for Object in root.findall( 'object'):
  11. bndbox=Object.find( 'bndbox')
  12. Size = root.find( 'size')
  13. value = (root.find( 'filename').text,
  14. int(Size.find( 'width').text),
  15. int(Size.find( 'height').text),
  16. Object.find( 'name'),
  17. int(bndbox.find( 'xmin').text),
  18. int(bndbox.find( 'ymin').text),
  19. int(bndbox.find( 'xmax').text),
  20. int(bndbox.find( 'ymax').text)
  21. )
  22. xml_list.append(value)
  23. column_name = [ 'filename', 'width', 'height', 'class', 'xmin', 'ymin', 'xmax', 'ymax']
  24. xml_df = pd.DataFrame(xml_list, columns=column_name)
  25. return xml_df
  26. def main():
  27. image_path = "D:/object_detection/research/object_detection/ROS1/VOCdevkit/RSDS2016/Annotations"
  28. xml_df = xml_to_csv(image_path)
  29. xml_df.to_csv( 'D:/object_detection/research/object_detection/ROS1/VOCdevkit/RSDS2016/labels.csv', index= None)
  30. print( 'Successfully converted xml to csv.')
  31. main()

注意更改程序中的路径,生成csv格式的文件。

(2)读取csv格式的文件,并选择数据集中的某一张图片,进行显示。

 

另外也可以在labelImg中加载图片和对应的标签,显示。

 

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值