logo
publist
写文章

简介

该用户还未填写简介

擅长的技术栈

可提供的服务

暂无可提供的服务

yolov5训练时遇到问题:ValueError: zero-size array to reduction operation maximum which has no identity

1、找了资料说是标签不匹配,需要进行标签归一化。2、推断出是标签问题,进行标签检查,发现在data/ xxx。yaml文件中的标签“cat” 打成了“car”。

#python
YOLOv5使用GPU训练

前提:yolov5在CPU训练可以进行后,想改用GPU提速遇到问题。原因:自己的torch下的是CPU版本。方法:重新创建一个解释器。step1:conda remove -n yolo --allstep2:conda create -n yolov5 python=3.8step3: 激活解释器,去PyTorch官网下载对应GPU版本的torch(用pip下载)step4:把yolov5文件下

#python#pytorch
到底了