from lxml import etree
import cv2
# 读取 xml 文件信息,并返回字典形式
def parse_xml_to_dict(xml):
if len(xml) == 0: # 遍历到底层,直接返回 tag对应的信息
return {xml.tag: xml.text}
result = {}
for child in xml:
child_result = parse_xml_to_dict(child) # 递归遍历标签信息
if child.tag != 'object':
result[child.tag] = child_result[child.tag]
else:
if child.tag not in result: # 因为object可能有多个,所以需要放入列表里
result[child.tag] = []
result[child.tag].append(child_result[child.tag])
return {xml.tag: result}
# xml 标注文件的可视化
def xmlShow(img, xml, save=True):
image = cv2.imread(img)
with open(xml, encoding='gb18030', errors='ignore') as fid: # 防止出现非法字符报错
xml_str = fid.read()
xml = etree.fromstring(xml_str)
data = parse_xml_to_dict(xml)["annotation"] # 读取 xml文件信息
ob = [] # 存放目标信息
for i in data['object']: # 提取检测框
name = str(i['name']) # 检测的目标类别
bbox = i['bndbox']
xmin = int(bbox['xmin'])
ymin = int(bbox['ymin'])
xmax = int(bbox['xmax'])
ymax = int(bbox['ymax'])
tmp = [name, xmin, ymin, xmax, ymax] # 单个检测框
ob.append(tmp)
# 绘制检测框
for name, x1, y1, x2, y2 in ob:
cv2.rectangle(image, (x1, y1), (x2, y2), color=(255, 0, 0), thickness=2) # 绘制矩形框
cv2.putText(image, name, (x1, y1 - 10), fontFace=cv2.FONT_HERSHEY_SIMPLEX,
fontScale=0.5, thickness=1, color=(0, 0, 255))
# 保存图像
if save:
cv2.imwrite('result.png', image)
# 展示图像
cv2.imshow('test', image)
cv2.waitKey()
cv2.destroyAllWindows()
if __name__ == "__main__":
import os
import random
img_path = './data/train/images/' # 传入图片
image = [os.path.join(img_path, i) for i in os.listdir(img_path)]
r = random.randint(0, len(image) - 1) # 生成随机索引
image_path = image[r]
labels_path = image_path.replace('images', 'labels') # 自动获取对应的 xml 标注文件
labels_path = labels_path.replace('.jpg', '.xml')
xmlShow(img=image_path, xml=labels_path, save=True)
没有合适的资源?快使用搜索试试~ 我知道了~
温馨提示
项目包含垃圾分类检测数据集(46分类)(VOC标注格式的xml文件),数据保存按照文件夹保存,可直接用作目标检测数据集,无需额外处理。 数据集介绍:数据集包含生活中常见的垃圾,例如塑料袋、空瓶子、金属罐头、书包、果皮等等共46个类别 数据总大小:237MB 数据分为分为训练集和测试集(data--train(test)----images(labels)) 训练集:3753张图片和3753个标签xml文件组成 测试集:938张图片和938个标签xml文件组成 ***除此外,包含检测类别的json字典文件*** 为了方便查看数据,提供了可视化py文件,随机传入一张图片即可绘制边界框,并且保存在当前目录。脚本无需更改,可以直接运行!
资源推荐
资源详情
资源评论
收起资源包目录
深度学习之目标检测数据集:垃圾分类检测数据集(46分类)(VOC标注格式的xml文件) (2000个子文件)
show.py 2KB
img_2640.xml 11KB
img_2296.xml 8KB
img_v2_4354.xml 4KB
img_v2_4358.xml 3KB
img_v2_972.xml 3KB
img_v2_4982.xml 3KB
img_v2_1060.xml 3KB
img_v2_5356.xml 3KB
img_v2_3703.xml 3KB
img_v2_4357.xml 3KB
img_v2_1066.xml 3KB
img_v2_4353.xml 3KB
img_v2_4352.xml 3KB
img_v2_5569.xml 3KB
img_v2_5014.xml 3KB
img_v2_3960.xml 3KB
img_000690.xml 3KB
img_v2_1190.xml 3KB
img_v2_1034.xml 3KB
img_v2_1408.xml 3KB
img_v2_4367.xml 3KB
img_v2_5177.xml 3KB
img_v2_962.xml 3KB
img_v2_119.xml 2KB
img_v2_1021.xml 2KB
img_v2_5897.xml 2KB
img_v2_4375.xml 2KB
img_v2_5996.xml 2KB
img_744.xml 2KB
img_v2_70.xml 2KB
img_v2_4870.xml 2KB
img_v2_4873.xml 2KB
img_v2_5220.xml 2KB
img_v2_702.xml 2KB
img_v2_4986.xml 2KB
img_12264.xml 2KB
img_v2_4993.xml 2KB
img_v2_3493.xml 2KB
img_v2_754.xml 2KB
img_v2_1047.xml 2KB
img_v2_1041.xml 2KB
img_v2_4965.xml 2KB
img_v2_1084.xml 2KB
img_v2_930.xml 2KB
img_v2_5091.xml 2KB
img_v2_1069.xml 2KB
img_v2_4884.xml 2KB
img_v2_1059.xml 2KB
img_v2_4908.xml 2KB
img_v2_4865.xml 2KB
img_v2_4898.xml 2KB
img_v2_5938.xml 2KB
img_v2_3252.xml 2KB
img_v2_4951.xml 2KB
img_v2_5787.xml 2KB
img_v2_3290.xml 2KB
img_v2_6005.xml 2KB
img_v2_4868.xml 2KB
img_v2_4392.xml 2KB
img_v2_3241.xml 2KB
img_v2_72.xml 2KB
img_739.xml 2KB
img_733.xml 2KB
img_v2_1903.xml 2KB
img_v2_3257.xml 2KB
img_v2_4411.xml 2KB
img_v2_3203.xml 2KB
img_v2_1798.xml 2KB
img_v2_2484.xml 2KB
img_v2_986.xml 2KB
img_v2_1845.xml 2KB
img_v2_818.xml 2KB
img_v2_4142.xml 2KB
img_v2_4460.xml 2KB
img_v2_943.xml 2KB
img_v2_4587.xml 2KB
img_v2_2626.xml 2KB
img_v2_954.xml 1KB
img_v2_1072.xml 1KB
img_v2_107.xml 1KB
img_v2_1320.xml 1KB
img_v2_1301.xml 1KB
img_v2_114.xml 1KB
img_v2_4885.xml 1KB
img_v2_4880.xml 1KB
img_v2_4160.xml 1KB
img_v2_4119.xml 1KB
img_v2_4861.xml 1KB
img_v2_18.xml 1KB
img_v2_4126.xml 1KB
img_v2_5046.xml 1KB
img_v2_4876.xml 1KB
img_v2_2625.xml 1KB
img_v2_3841.xml 1KB
img_v2_151.xml 1KB
img_v2_1795.xml 1KB
img_v2_4218.xml 1KB
img_v2_5361.xml 1KB
img_v2_911.xml 1KB
共 2000 条
- 1
- 2
- 3
- 4
- 5
- 6
- 20
资源评论
听风吹等浪起
- 粉丝: 1w+
- 资源: 1012
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功