当前位置:   article > 正文

YOLOv5运行delect.py文件时出现ModuleNotFoundError: No module named ‘torchvision‘的解决办法_modulenotfounderror: no module named 'torchvision

modulenotfounderror: no module named 'torchvision

1、问题:

在anaconda安装了pytorch后,用delect.py打开本机摄像头时出现报错:

ModuleNotFoundError: No module named 'torchvision'

且pycharm提示安装包不符合要求。

(不符合要求的安装包分别是opencv-python、torch、torchvision、thop)

且按照提示安装相关包时出现以下报错:

To search for alternate channels that may provide the conda package you’re looking for, navigate to

https://anaconda.org

and use the search bar at the top of the page.

2、解决方法1:

更新torchvision的版本至0.4.1

  1. pip install cffi -i http://pypi.douban.com/simple --trusted-host pypi.douban.com
  2. pip install cmake -i http://pypi.douban.com/simple --trusted-host pypi.douban.com
  3. pip install mkl -i http://pypi.douban.com/simple --trusted-host pypi.douban.com
  4. pip install pyyaml -i http://pypi.douban.com/simple --trusted-host pypi.douban.com
  5. pip install numpy -i http://pypi.douban.com/simple --trusted-host pypi.douban.com
  6. pip install torchvision==0.4.1 -i http://pypi.douban.com/simple --trusted-host pypi.douban.com

版权声明:以上来自CSDN博主「叶庭云」
原文链接:https://blog.csdn.net/fyfugoyfa/article/details/115800768

在anaconda执行以上命令后,出现已安装过该安装包的提示。

故尝试方法2.

3、解决方法2:

在anaconda中查看pytorch安装包

conda list

 显示列表已有torchvision,故我们需先卸载已有pytorch。

在anaconda中执行以下命令行以卸载原有pytorch(在cmd中执行也可以):

  1. pip uninstall torchvision
  2. pip uninstall torch

卸载完毕后,在Start Locally | PyTorchhttps://pytorch.org/get-started/locally/选择合适自己的版本,复制官方给出的命令行,在anaconda执行安装pytorch

安装完毕后,发现pycharm不再报错:ModuleNotFoundError: No module named 'torchvision'

但是不符合要求的安装还有opencv-python 和thop

于是在anaconda或python终端上执行命令行:

  1. pip install thop
  2. pip install opencv-python

执行后python安装这两个包。安装完毕后不再报错。

4、打开本机摄像头运行YOLOv5自带的图像识别模型:

原理解释:YOLOv5的delect.py默认的是识别数据包中自带的两张图片,如果直接执行delect.py 将会创建一个runs文件夹,在其中的exep文件中会出现识别完毕后的两张图片(以下是图片bus.jpg):

 如果我们想打开自己的本机摄像头则需要改变delect.py中的参数,操作可参考这个博主的做法在yolov5项目中如何使用自带摄像机不用网络摄像机进行实时检测?_风吹落叶的博客-CSDN博客

声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:【wpsshop博客】
推荐阅读
相关标签
  

闽ICP备14008679号