耗时56个小时,硬件为:双卡 Tesla V100 32GB显存版,自己看结果,5X性能居然不如5S,理论上5X训练慢,FPS低,但是准确率很高,但是从结果看,还不如5S,所以后续的工作:
1、试一试性价比更高的5M 5L
2、尝试进行数据增强,这里建议训练的时候,修改增大分辨率
我这还没有进行调参、数据预处理,全是默认参数,YOLO5S的性能跳到了0.93,下一步,打算对yolo5S进行性能分析,测试集由于没有label,找不到原因
那我们就从训练集开始总结吧,把预测错误的原因找出来,抓出主要矛盾!