为了训练自己的数据集,需要将数据集转化为voc格式。通过以下6个步骤来实现。
1.数据增强
通过几种扩充方式来增加数据,如亮度增强、对比度增强、旋转、翻转和加入噪声(每种方法在最后均需将图片resize成300*300的大小)。
代码:improve.py
2.标记
使用labelme进行标注,得到json文件,也可以通过labelmg标注,直接可得到xml文件。标注信息包括目标类别与包围框信息。
3.读取标注图像并展示
根据使用自定义txt文件读取数据、手动读取自己的数据来展示,但是我遇到了这个问题:
AttributeError: module ‘labelme.utils’ has no attribute ‘draw_label’
然后通过解决办法以及coco标注信息与labelme标注信息的详解、相互转换及可视化来解决。
代码:json_read.py
4.po
个人数据集制作、转化为voc格式并融合
最新推荐文章于 2023-02-12 12:16:42 发布