from lxml import etree
import cv2
# https://blog.csdn.net/qq_44886601/article/details/135400117
# 读取 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)
没有合适的资源?快使用搜索试试~ 我知道了~
温馨提示
项目包含:二维码目标检测数据集(VOC标注格式的xml文件,已经做了训练集和测试集划分,数据保存按照文件夹保存,经测试,可直接用作目标检测数据集,无需额外处理。 图像分辨率为300-800的RGB图片,数据生活中常见二维码图标 【数据集介绍】二维码图像数据,1类别:QR二维码 【数据总大小】125MB 【数据集详情】data目录下分为两个目录,train为训练数据、test为测试数据,每个目录下各有两个子文件夹。images存放数据图片,labels目录存放标注文件。训练集共1492张jpg图片和1492个xml解释文件。训练集共372张jpg图片和372个xml解释文件。除此外还有1200张未标注的测试图像 【json文件】1类别的json字典文件 为了方便查看数据,提供了可视化py文件,随机传入一张图片即可绘制边界框,并且保存在当前目录。脚本无需更改,可以直接运行!
资源推荐
资源详情
资源评论
收起资源包目录
目标检测数据集:二维码目标检测数据集(VOC标注格式的xml文件,已经做了训练集和测试集划分 (2000个子文件)
QR-00334.jpg 349KB
QR-00164.jpg 311KB
QR-00140.jpg 288KB
QR-01879.jpg 234KB
QR-00362.jpg 199KB
QR-00734.jpg 196KB
QR-01100.jpg 184KB
QR-00387.jpg 168KB
QR-00135.jpg 166KB
QR-00169.jpg 157KB
QR-01825.jpg 153KB
QR-00448.jpg 149KB
QR-00752.jpg 143KB
QR-00943.jpg 130KB
QR-00500.jpg 120KB
QR-00817.jpg 118KB
QR-00674.jpg 118KB
QR-00458.jpg 111KB
QR-00657.jpg 93KB
QR-01507.jpg 88KB
QR-00856.jpg 88KB
QR-00774.jpg 88KB
QR-00749.jpg 77KB
QR-00695.jpg 73KB
QR-00353.jpg 71KB
QR-00004.jpg 69KB
QR-00371.jpg 68KB
QR-00987.jpg 67KB
QR-00357.jpg 65KB
QR-00478.jpg 64KB
QR-00988.jpg 62KB
QR-00842.jpg 59KB
QR-00279.jpg 59KB
QR-01616.jpg 56KB
QR-00239.jpg 55KB
QR-00400.jpg 55KB
QR-01051.jpg 54KB
QR-00285.jpg 54KB
QR-01521.jpg 54KB
QR-00845.jpg 54KB
QR-01042.jpg 54KB
QR-00656.jpg 54KB
QR-00952.jpg 53KB
QR-00830.jpg 51KB
QR-00557.jpg 51KB
QR-01337.jpg 49KB
QR-01405.jpg 48KB
QR-00287.jpg 48KB
QR-01614.jpg 47KB
QR-01550.jpg 47KB
QR-01322.jpg 47KB
QR-01261.jpg 46KB
QR-01595.jpg 44KB
QR-01302.jpg 44KB
QR-01472.jpg 43KB
QR-00236.jpg 43KB
QR-00515.jpg 43KB
QR-00799.jpg 42KB
QR-00631.jpg 42KB
QR-00780.jpg 42KB
QR-01499.jpg 42KB
QR-01701.jpg 42KB
QR-01025.jpg 41KB
QR-01690.jpg 41KB
QR-01647.jpg 41KB
QR-01520.jpg 40KB
QR-00433.jpg 40KB
QR-00568.jpg 40KB
QR-00306.jpg 39KB
QR-00706.jpg 39KB
QR-01043.jpg 39KB
QR-01559.jpg 38KB
QR-01194.jpg 38KB
QR-00373.jpg 37KB
QR-00336.jpg 37KB
QR-01545.jpg 36KB
QR-00530.jpg 36KB
QR-01062.jpg 36KB
QR-00964.jpg 35KB
QR-00111.jpg 35KB
QR-01327.jpg 35KB
QR-01379.jpg 34KB
QR-00577.jpg 34KB
QR-01582.jpg 34KB
QR-01565.jpg 34KB
QR-00556.jpg 34KB
QR-01496.jpg 34KB
QR-00664.jpg 33KB
QR-01794.jpg 33KB
QR-00049.jpg 33KB
QR-00976.jpg 31KB
QR-00085.jpg 31KB
QR-00601.jpg 31KB
QR-00991.jpg 30KB
QR-01275.jpg 30KB
QR-01273.jpg 29KB
QR-00517.jpg 29KB
QR-00325.jpg 28KB
QR-00628.jpg 27KB
QR-00901.jpg 27KB
共 2000 条
- 1
- 2
- 3
- 4
- 5
- 6
- 20
资源评论
听风吹等浪起
- 粉丝: 1w+
- 资源: 1070
下载权益
C知道特权
VIP文章
课程特权
开通VIP
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功