YOLO系列检测时没有预测框输出的解决办法

问题描述

在训练好YOLO模型之后，运行detect.py文件发现输出，输入的图像没有框，也没有label的txt文件输出。

原因分析

• YOLO读取权重错误。在detect.py文件里面的权重路径错误，读取不是yolo的权重。
• 在detect.py文件内缺少

cudnn.benchmark = True

• Pytorch版本错误

解决办法

• 修改权重
• 在else下面插入

cudnn.benchmark = True

• 修改Torch版本，如果版本不对应会导致预测值全为Nan，就不能输出预测框。