一、目标检测数据集制作:
先安装labelimg,pip install labelimg即可。
- 创建下图中所示的3个文件夹,photo存放图片的文件夹
- Labelimg—>open Dir打开photo的文件夹;利用Change Save Dir把标签文件指定存放在Annotations(注:打开labelimg——>View—>Auto Save mode,自动将标签保存到已经选择的位置),点击CreateRectBox绘制框,绘制完框后,输入种类。
(1)标注的物体触及到底端的时候,从下往上标注。 - 将photo里面的图片复制到JPEGImages
二、训练VOC数据集出现的问题记录
- TypeError: init() got an unexpected keyword argument ‘serialized_options’
运行这句代码解决了该问题,但是又出现了其他问题 - 运行pip install future解决下面的问题。
- 下面的错误是因为GPU的算力不够,pytorch版本较高