093自动生成xml目标检测框数据集-不用手动标注

本文介绍了如何使用Python脚本自动识别图像中的轮廓,通过调整阈值并生成XML文件,用于目标检测模型的训练,显著节省人工标注时间。
摘要由CSDN通过智能技术生成

直接上效果图:

这个是通过轮廓查找,自动将检测的到的轮廓外接矩形保存成xml文件,可以供目标检测数据集训练使用。

通过自动标注可以节省自己大量的时间和精力。

代码下载和视频演示地址:

093自动生成xml目标检测框数据集-不用手动标注_哔哩哔哩_bilibili

回到代码部分,整个代码分为两个部分,(一个文件夹存放图片和两个py代码文件)

一个部分是,运行第一个py文件,

命令行输入 python 01Determine_threshold.py

,弹出可视化界面,然后我们加载需要处理的图片,再选择合适的轮廓查找阈值。

第二部分就是自动运行生成xml文件部分

首先打开02auto_xml.py文件,将里面第71行的阈值修改成上面获取得到的最合适的阈值,

然后再命令行输入

python 02auto_xml.py

就会将jpg文件夹里面的所有图片都进行轮廓查找,将外接矩形框保存成xml文件。

该生成的xml文件可以用labelimg打开检查生成效果,可用于目标检测数据集训练模型。

  • 7
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值