Yolov5 anaconda常见问题

一:cpu版本

1. 无法安装opencv-python

改为opencv即可

2. 没有thop

官网下载安装包GitHub - Lyken17/pytorch-OpCounter: Count the MACs / FLOPs of your PyTorch model.

解压后在sitepackage里安装

---路径输入cmd 同时按shift和鼠标右键即可打开当前路径下的终端

python setup.py install

 参考:【YOLOv5问题记录】thop库的安装_啥都不会的材料菜鸡的博客-CSDN博客

3. ValueError: The requested array has an inhomogeneous shape after 1 dimensions

先debug到出问题那行,接着看shape,找到值none对应的图片(也就是出错的那张图),再到训练集文件夹中把对应图片的名字修改一下,重新运行即可解决问题。

参考:YOLOv5训练过程中遇到该问题的解决方法ValueError: The requested array has an inhomogeneous shape after 1 dimensions_Z_U_Z_U的博客-CSDN博客

二. cpu转gpu

1. torch torchvision下载相关(根据cuda版本 )

https://download.pytorch.org/whl/torch_stable.html

类似步骤2,进行安装,需要注意的是:

1). conda activate 相应的环境

2). cd到安装包目录

3). python -m pip install 包名(tab补全)

---否则pycharm 难以识别新的torch 和torchvision 包

2. TypeError: can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.

将报错代码self.numpy()改为self.cpu().numpy()即可 

参考:解决“TypeError: can‘t convert cuda:0 device type tensor to numpy. ......”问题_can't convert cuda:0 device_Zpadger的博客-CSDN博客

其他:

解决Yolov5运行时,报错问题— assert img0 is not None, ‘Image Not Found ‘ + pathAssertionError: Image Not Found_mob604756fb6267的技术博客_51CTO博客

yolov5踩坑系列-1(error:‘Hardswish‘ object has no attribute ‘inplace‘)_ynxdb2002的博客-CSDN博客

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值