使用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

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

基于深度学习的农作物叶片病害检测系统（UI界面+YOLOv5+训练数据集）
摘要:农作物叶片病害检测系统用于智能检测常见农作物叶片病害情况,自动化标注.记录和保存病害位置和类型,辅助作物病害防治以增加产值.本文详细介绍基于YOLOv5深度学习模型的农作物叶片病害检测系统,在介 ...
基于深度学习的瓶子检测软件（UI界面+YOLOv5+训练数据集）
摘要:基于深度学习的瓶子检测软件用于自动化瓶子检测与识别,对于各种场景下的塑料瓶.玻璃瓶等进行检测并计数,辅助计算机瓶子生产回收等工序.本文详细介绍深度学习的瓶子检测软件,在介绍算法原理的同时,给出P ...
水果新鲜程度检测系统（UI界面+YOLOv5+训练数据集）
摘要:水果新鲜程度检测软件用于检测水果新鲜程度,利用深度学习技术识别腐败或损坏的水果,以辅助挑拣出新鲜水果,支持实时在线检测.本文详细介绍水果新鲜程度检测系统,在介绍算法原理的同时,给出Python的 ...
解决YOLOV5训练时P、R、mAP等值均为0的问题
最近用YOLOV5训练自己的数据集,出现了训练失败的情况,比如box,obj,cls,labels等均为nan或0,找了很多办法,其实就是cuda与PyTorch版本的问题 Epoch gpu_mem ...
批量下载百度搜索图片+labelimg制作自己的数据集+转换至Yolo-v5训练数据集
由于课题需要,需要自己制作数据集进行训练,目前是自己制作的第二个数据集,发现有某些细节已经忘记,记录备忘,同时为后来者提供借鉴.文章以car-tank数据集做为例子介绍整体流程: 1.准备数据:从各 ...
YOLOV5训练数据集过程中特殊问题记录
项目场景: yolov5训练GX数据集问题描述: 运行train.py Traceback (most recent call last):File "/home/milk/yolov52 ...
Yolo实用指南（step by step）之三yolov5训练自己的数据集
关于yolov5训练自己的数据集的文章网上已经有一大把,自己之前也摸索了数据标注有一阵子,可一直迟迟未动手,感觉对原理不熟悉,很多东西浮光掠影有些肤浅,思来想去还是花了整整一天实践一下吧,毕竟这是基础 ...
YOLOV5训练自己的数据集（踩坑经验之谈）
导言很惭愧,来csdn已经三年多了,却一直都在"白嫖"各位大神的经验与总结.这几天也一直在csdn里学习YOLOv3与YOLOv5训练数据集的具体步骤,几经波折终于实现了很好的效 ...
YOLOv5训练数据提示No labels found、with_suffix使用、yolov5训练时出现WARNING: Ignoring corrupted image and/or label
YOLOv5训练数据提示No labels found[亲测yolo加载label文件只需要修改img2label_paths函数改变就可以加载到label文件了] 仔细看下数据加载.处理的文件dat ...

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

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

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

最新文章

热门文章