本文以yolov5-7.0的代码进行讲解yolov5的改进技巧。
1. 使用yolov5-p6以及yolov5-p7模型
yolov5 官方实现的代码，提供了多种架构的模型配置文件，包括:yolov3-spp,yolov5-bifpn, yolov5-p6, yolov5-p7, yolov5-panet, yolov5-transformer
如果想训练不同架构的模型，只需要在train.py脚本中, 将--cfg(模型配置yaml),和--weights 修改为对应模型的yaml文件路径，以及预训练weights路径，就可以进行训练，执行命令如:
python train.py -