当前位置:   article > 正文

pycharm运行yolov5-v5.0 (深度学习yolov5-篇二)_pycharm yolov5

pycharm yolov5

这一篇记录使用pycharm运行yolov5,有关于yolov5环境搭建请看本栏的第一篇文章(是使用Anaconda搭建的cpu虚拟环境)。

一. 从官网下载yolov5项目

yolov5-v5.0链接

二、pycharm打开yolov5项目

打开这个项目

进来之后会弹出虚拟环境创建的窗口,我们已经再anaconda中创建了我们自己的虚拟环境了,所以不用在这里创建。这里选择离开。

1、配置虚拟环境解释器

点击file -> 设置选项

点击python Interpreter为当前项目配置解释器。

点击添加解释器

然后选择系统环境,再选择anaconda 安装目录下的虚拟环境

!!!注意这里选择的是该虚拟环境中的python.exe文件。

然后点击OK,这里可以看到该环境中的一下配置库,例如pytorch pip numpy...

接着点击Apply->ok等待解释器更新完成。

2、下载配置包

更新完成之后我们找到 requirements.txt文件,注释掉其中的torch、torchvision这两行,因为这个我们再pytorch中已经自己安装了,所以不需要重复安装。

然后再终端运行pip install -r requirements.txt 命令下载需要用的包,等待下载完成即可。

pip install -r requirements.txt

三. 运行detect.py查看检测结果

找到detect.py文件右击运行。

运行之后日志中可能还会报一些模块未找到,例如pandas、

我们可以在anaconda中去下载这些对应的模块

例如上述这个报错,我们使用 pip install pandas 命令。

然后回到pycharm等待更新完成,我们再次运行。如果还有其它模块也是一样的使用命令去下载即可,

同时也可以在pycharm的终端去下载,命令是一样的。 (pip install tqdm)

都下载完成后会下载yolov5s.pt权重文件

后面就会输出检测结果了。会显示检测结果的保存路径以及检测的图像信息集检测信息。

我们去找到日志显示的这个文件夹。

这里就是检测的结果图像了。

四. 修改yolo参数

1、修改检测图像

如果我们要测试检测其他图像可以在data/images 目录下添加图片

那么到这里就可以完美运行官网yolov5-v5.0的项目了。

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/知新_RL/article/detail/202084
推荐阅读
相关标签
  

闽ICP备14008679号