直接上效果图:
这个是通过轮廓查找,自动将检测的到的轮廓外接矩形保存成xml文件,可以供目标检测数据集训练使用。
通过自动标注可以节省自己大量的时间和精力。
视频演示和demo仓库地址:
093自动生成xml目标检测框数据集-不用手动标注_哔哩哔哩_bilibili
回到代码部分,整个代码分为两个部分,(一个文件夹存放图片和两个py代码文件)
一个部分是,运行第一个py文件,
命令行输入 python 01Determine_threshold.py
,弹出可视化界面,然后我们加载需要处理的图片,再选择合适的轮廓查找阈值。
第二部分就是自动运行生成xml文件部分
首先打开02auto_xml.py文件,将里面第71行的阈值修改成上面获取得到的最合适的阈值,
然后再命令行输入
python 02auto_xml.py
就会将jpg文件夹里面的所有图片都进行轮廓查找,将外接矩形框保存成xml文件。
该生成的xml文件可以用labelimg打开检查生成效果,可用于目标检测数据集训练模型。