YOLOv5运行delect.py文件时出现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

pip install cffi -i http://pypi.douban.com/simple --trusted-host pypi.douban.com
pip install cmake -i http://pypi.douban.com/simple --trusted-host pypi.douban.com
pip install mkl -i http://pypi.douban.com/simple --trusted-host pypi.douban.com
pip install pyyaml -i http://pypi.douban.com/simple --trusted-host pypi.douban.com
pip install numpy -i http://pypi.douban.com/simple --trusted-host pypi.douban.com

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

YOLOv5运行delect.py文件时出现ModuleNotFoundError: No module named ‘torchvision‘的解决办法_第1张图片

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

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

pip uninstall torchvision
pip uninstall torch

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

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

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

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

pip install thop
pip install opencv-python

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

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

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

YOLOv5运行delect.py文件时出现ModuleNotFoundError: No module named ‘torchvision‘的解决办法_第2张图片

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

你可能感兴趣的:(pycharm,pytorch,python,深度学习)