yolov5剪枝实战4: 正常训练和稀疏化训练
下,我们可以看到训练过程中产生的中间文件,包括result.png可以看到训练过程中的loss以及mAP指标的变化趋势,另外在weights保持best.pt和last.pt,best.pt对应我们训练出来性能最好的模型文件,last.pt对应最后一个epoch训练保存的权重文件。创建好了voc_ball.yaml 和yolov5s_ball.yaml这两个文件后,就有了针对自己数据集的配置文件了。针对自己的数据集,需要训练用的配置文件。使用yolov5s训练处的权重。使用yolov5s训练出的权重。
复制链接