训练voc数据集的步骤：
- ①：首先下载voc数据集，将数据集放在从github中下载的项目中VOCdevkit目录中（直接将数据集拉入到项目中，替代目标文件即可）
源码下载：
https://github.com/bubbliiiing/yolo3-pytorch
(22条消息) 睿智的目标检测26——Pytorch搭建yolo3目标检测平台_Bubbliiiing的学习小课堂-CSDN博客_睿智的目标检测26
下载的目录如下，其中Annotations为标签文件，JPEGImages存放的为图片文件，ImageSets目录下存放的为.txt文件，用于训练和验证。
- ②：将voc_annotation.py中的annotation_mode = 2设置为2，直接在根目录下生成2007_train.txt和22007_val.txt文件用于训练和验证，完成两个文件的生成之后打开train.py文件
- ③：其中train.py文件中的参数都是为了训练voc数据集进行设置的。
classes_path = 'model_data/voc_classes.txt'#将classes_path设置为'model_data/voc_classes.txt'
设置完成之后，可以直接运行train.py对voc数据集进行训练
训练自己数据集的步骤：
- 首先需要利用labelme完成数据集的制作，其中制作的数据集需要包含两个部分一个是其中Annotations为标签文件，JPEGImages存放的为图片文件.
- 然后运行根目录下的voc_annotation.py文件，用于生成训练需要的txt文件，第一部分的txt是JPEGImages中的txt用于，用于划分测试集，验证集和训练集，第二部分是根目录下的2007_train.txt和2007_val.txt，用于模型的训练
- 第一次训练，要修改train.py中的一个参数，即classes_path = 'model_data/类别.txt'，其中类别.txt中是我们自己制作的区分类别，例如区分猫和狗，里面的内容为dog cat，每个各占一行。第一次训练只需要修改classes_path中的参数
- train.py文件与voc_annotation.py文件的classes_path的参数需要保持一致。
利用训练好的模型进行预测的步骤：
需要修改根目录下的yolo.py和predict.py文件中的参数。
#--------------------------------------------------------------------------# # 使用自己训练好的模型进行预测一定要修改model_path和classes_path！ # model_path指向logs文件夹下的权值文件，classes_path指向model_data下的txt # # 训练好后logs文件夹下存在多个权值文件，选择验证集损失较低的即可。 # 验证集损失较低不代表mAP较高，仅代表该权值在验证集上泛化性能较好。 # 如果出现shape不匹配，同时要注意训练时的model_path和classes_path参数的修改 #--------------------------------------------------------------------------# "model_path" : 'logs/yolo_weights.pth', "classes_path" : 'model_data/类别.txt',input_shape必须是32的倍数，yolo.py中的classes_path与train.py中的需要保持一致
修改之后，运行predict.py文件，进行预测。