栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > C/C++/C#

使用PyCharm及其等价命令行运行YoloV5 (启用GPU加速)

C/C++/C# 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

使用PyCharm及其等价命令行运行YoloV5 (启用GPU加速)

目录

前提

1.copy过来后要删除其根目录下的.idea folder:

2.删除下面folder里面的所有内容:

3.在Pycharm中重新建立一个Virtual enviorenment:

4.在Pycharm的终端安装YoloV5 的requirements:

5.可能会遇到下面下面问题,解决方案见下:

6.如果你的机器有GPU显卡,则可按如下启用GPU加速:

另外,可用pyCharm的等价命令行启动Yolo V5:


前提

系统环境:Windows10.

首先安装Windows 64bit 的Python 3.6.2。

再安装PyCharm的社区版.

然后从官网Clone Yolo V5.

我是从从别处copy的整个Yolo v5 整个folder,如果不是copy的source code ,从下面第3步开始:

1.copy过来后要删除其根目录下的.idea folder:

因为这个folder会留有以前的环境信息,导致run不起来:

    C:XXXyolov5.ideaworkspace.xml (

        行 54:      

2.删除下面folder里面的所有内容:

    C:XXXyolov5venv

如果不是copy的source code ,不用做以上两步。

3.在Pycharm中重新建立一个Virtual enviorenment:

4.在Pycharm的终端安装YoloV5 的requirements:

5.可能会遇到下面下面问题,解决方案见下:

 building 'Levenshtein._levenshtein' extension

  error: Microsoft Visual C++ 14.0 or greater is required. Get it with "Microsoft C++ Build Tools": Microsoft C++ Build Tools - Visual Studio

请参考下面连接解决,实际是需要安装C++ build工具:

error: Microsoft Visual C++ 14.0 or greater is required. Get it with "Microsoft C++ Build Tools": https://visualstudio.microsoft.com/visual-cpp-build-tools/ - Microsoft Q&A

6.如果你的机器有GPU显卡,则可按如下启用GPU加速:

假如你有GPU显卡,先跑一下yolov 5的测试,如果出现如下错误,则说明可能pytorch未安装正确:

            AssertionError: CUDA unavailable, invalid device 0 requested

解决如下,在Pycharm的终端安装:

pip3 install torch==1.10.0+cu102 torchvision==0.11.1+cu102 torchaudio===0.10.0+cu102 -f https://download.pytorch.org/whl/cu102/torch_stable.html

上面命令行来自 ,如下:

如上步骤,则可以使用Pycharm run Yolo V5的detect.py,记得设置该文件参数加上--device 0,表示启用GPU. 

另外,可用pyCharm的等价命令行启动Yolo V5:

先在键盘上按Win+R 启动终端cmd:

直接进入yolov5下面的folder,以激活PyCharm的虚拟环境 :

C:XXXyolov5venvscripts>activate.bat

(venv) C:XXXyolov5venvscripts>cd ..

假如你用了Mark Directory as Sources Root:

那么请使用以下,达到等价效果:

(venv) C:XXXyolov5>set PYTHonPATH=%PYTHONPATH%;C:XXXyolov5PaddleOCR

检测一下上面的设置:

(venv) C:XXXyolov5>echo %PYTHONPATH%

%PYTHONPATH%;C:XXXyolov5PaddleOCR

最后命令行启动Yolo V5的测试:

(venv) C:XXXyolov5>python detect.py --weights yolov5s.pt --source 0 --device 0

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/352653.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号