0


【YOLOv5问题记录】thop库的安装

最近开始学习YOLOv5,踩了不少坑,总结一下问题。

配置环境按照这篇教程来的:Yolov5的配置+训练(超级详细!!!)_小学生玩编程的博客-CSDN博客

训练数据集跟着炮哥的这篇:目标检测---教你利用yolov5训练自己的目标检测模型_yolov5如何训练自己的模型_炮哥带你学的博客-CSDN博客


问题描述:

环境配置完,在跟这炮哥教程训练数据集的时候,Pycharm终端运行以下命令

pip install -r requirements.txt

此时出现了pycocotools,opencv-python和thop这三个库的安装问题,出现提示:

不满足软件包要求' Opencv- python>=4.12,thop'' pycocotools>=2.0


分析:

就是库的安装出现了问题,所以尝试单独安装thop库

  1. 直接在annaconda prompt用各种安装方法,例如:
pip install thop
conda install thop

都不管用,都会出现error

  1. 当时报错的信息建议我们去www.anaconda.org下载

pycocotools库就是用这种方法解决的,但是thop似乎在网站里搜不到


thop库的解决方法:

Step1:找到 thop 包的github官方链接

GitHub - Lyken17/pytorch-OpCounter: Count the MACs / FLOPs of your PyTorch model.

Step2:下载压缩文件,解压到虚拟环境的site-packages文件下

我的虚拟环境叫pytorch

对应路径是 D:\Applications\anaconda3\envs\pytorch\Lib\site-packages

Step3:打开annaconda prompt,激活进入自己的虚拟环境

env_name改为你的的虚拟环境名字

conda activate env_name

Step4:cd到压缩包解压的路径下

注意这里cd到路径下似乎只能一个一个来

直接cd D:\Applications\anaconda3\envs\pytorch\Lib\site-packages似乎不管用

cd..  //返回到上一个文件夹
cd /d d:  //从C盘切换到D盘

7b3f8269acb247fb92cf3a6f3e3e8878.png

上面我有手残输错的请忽略

(pytorch) C:\Users\26977>cd..

(pytorch) C:\Users>cd..

(pytorch) C:\>cd /d d:

(pytorch) D:\>cd Applications

(pytorch) D:\Applications>cd anaconda3

(pytorch) D:\Applications\anaconda3>cd envs

(pytorch) D:\Applications\anaconda3\envs>cd pytorch

(pytorch) D:\Applications\anaconda3\envs\pytorch>cd Lib

(pytorch) D:\Applications\anaconda3\envs\pytorch\Lib>cd site-packages

(pytorch) D:\Applications\anaconda3\envs\pytorch\Lib\site-packages>cd pytorch-OpCounter-master

Step5:在annaconda prompt输入命令python setup.py install,等待安装完成即可

7fd9d649fe62486aafa5e867e5841628.png

c34275bdecae42d18339b7da7e5505f4.png

出现Finished,回到Pycharm界面,不再提示thop库的问题,问题解决


本文转载自: https://blog.csdn.net/weixin_51972142/article/details/130514283
版权归原作者 啥都不会的材料菜鸡 所有, 如有侵权,请联系我们删除。

“【YOLOv5问题记录】thop库的安装”的评论:

还没有评论