yolov3自制数据集训练及预测方法

yolov3自制数据集预测方法

1、pip install labelimg
2、标注我们的数据。
3、拷贝到这个目录下。
yolo3-pytorch-mastertrain\VOC2007\Annotations

具体格式如下:

yolov3自制数据集训练及预测方法_第1张图片

4、写好所需的配置文件。
yolov3自制数据集训练及预测方法_第2张图片
安装git软件后,通过create_custom_model.sh文件点击右键后,点击git bash here,弹出窗口,输入:create_custom_model.sh 3(任务类别个数),软件自动生成文件:yolov3-custom.cfg。
yolov3-custom.cfg文件只改变配置:[yolo]中的classes=3;filters=3*(5+3)=24,完成。
yolov3自制数据集训练及预测方法_第3张图片
5、标签格式转换:
1、labelimg-x1, y1, x2, y2 YOLO-V3-Cx,Cy,W, H相对位置(0到1之间)
xml转换为txt
xmltotxt.py文件代码如下:


```python
# 开发时间&#

你可能感兴趣的:(python,yolov3,目标检测,pytorch,python)