YOLOV5训练数据集P、R、mAP等均为0的解决方案

使用YOLOV5训练数据集时,P、R等值均为0

最近在用YOLOV5训练自己的数据集,执行了十几个epochs之后,epoch的主要参数,比如box,obj,cls,labels等均为nan或0,即没有跑通正确的结果,找了很多方案,最终发现是pytorch版本不适合。

我的cuda版本是10.2,pytorch版本是1.7.1,重新安装了1.8.0版本的(pytorch-1.8.0-py3.8_cuda11.1_cudnn8_0.tar.bz2)以及相应的torch1.8.0(cu102/torch-1.8.0-cp38-cp38-win_amd64.whl)以及与之匹配的torchvision(torchvision-0.9.0-py38_cu102.tar.bz2)后,成功解决了这一问题。
torch和torchvison的匹配可以参考如下链接:https://codechina.csdn.net/mirrors/pytorch/vision?utm_source=csdn_github_accelerator

附pytorch下载链接https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/

torch和torchvision下载链接https://download.pytorch.org/whl/torch_stable.html

yolov5训练自己的数据集过程中还遇到一些其他问题,在我的另一篇博客有详细说明https://blog.csdn.net/weixin_42475780/article/details/121078534

来源:好乐无荒_

物联沃分享整理
物联沃-IOTWORD物联网 » YOLOV5训练数据集P、R、mAP等均为0的解决方案

发表评论