目标检测 | YOLOv3训练自己的数据全流程

1.构建YOLOv3网络的cfg文件

该文件表示的是你的检测网络的结构，类似caffe的prototxt文件。
YOLOv3的cfg文件
上篇介绍YOLOv3网络中提到的去掉上采样操作的YOLOv3cfg文件

2.准备hs.data文件

如下文件中规定了当前目标检测网络中的类别数量为4。训练数据的路径train.txt的位置，hs.names的路径的位置，以及最终训练得到的网络模型保存的位置。

classes = 4
train = /home/yuanlei/darknet_yolov3/backup/yolov3_initial/4_cls/train.txt
names = /home/yuanlei/darknet_yolov3/backup/yolov3_initial/4_cls/hs.names
backup = /home/yuanlei/darknet_yolov3/backup/yolov3_initial/4_cls

• train.txt文件
  train.txt文件中的内容如下图所示，保存的是训练数据集图片的保存路径。

/home/yuanlei/traindata_4cls/0.jpg
/home/yuanlei/traindata_4cls/1.jpg
/home/yuanlei/traindata_4cls/2.jpg
/home/yuanlei/traindata_4cls/3.jpg
...
...
/home/yuanlei/traindata_4cls/1000.jpg
/home/yuanlei/traindata_4cls/1001.jpg
...

• hs.names文件
  hs.names中的内容如下图所示，保存的是当前检测网络中需要检测的类别名称，用于后续显示。

people
car
bicycle
truck

3.训练数据准备

训练数据的保存位置要和步骤2中的train.txt文件中的路径对应起来，当然除了train.txt中的图片数据，我们还需要准备基于对应图片的目标位置和类别数据，也就是图片对