问题描述 在训练好YOLO模型之后,运行detect.py文件发现输出,输入的图像没有框,也没有label的txt文件输出。 原因分析 YOLO读取权重错误。在detect.py文件里面的权重路径错误,读取不是yolo的权重。在detect.py文件内缺少 cudnn.benchmark = True Pytorch版本错误 解决办法 修改权重在else下面插入 cudnn.benchmark = True 修改Torch版本,如果版本不对应会导致预测值全为Nan,就不能输出预测框。