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

一. 从官网下载yolov5项目

yolov5-v5.0链接




Yolov5_DeepSort_Pytorch部署 yolov5 pycharm_python


二、pycharm打开yolov5项目

打开这个项目


Yolov5_DeepSort_Pytorch部署 yolov5 pycharm_Powered by 金山文档_02


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


Yolov5_DeepSort_Pytorch部署 yolov5 pycharm_python_03


1、配置虚拟环境解释器

点击file -> 设置选项


Yolov5_DeepSort_Pytorch部署 yolov5 pycharm_虚拟环境_04


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


Yolov5_DeepSort_Pytorch部署 yolov5 pycharm_官网_05


点击添加解释器


Yolov5_DeepSort_Pytorch部署 yolov5 pycharm_虚拟环境_06


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


Yolov5_DeepSort_Pytorch部署 yolov5 pycharm_python_07


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


Yolov5_DeepSort_Pytorch部署 yolov5 pycharm_虚拟环境_08


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


Yolov5_DeepSort_Pytorch部署 yolov5 pycharm_虚拟环境_09


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

2、下载配置包

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


Yolov5_DeepSort_Pytorch部署 yolov5 pycharm_深度学习_10


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

pip install -r requirements.txt

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

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


Yolov5_DeepSort_Pytorch部署 yolov5 pycharm_python_11


Yolov5_DeepSort_Pytorch部署 yolov5 pycharm_虚拟环境_12


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


Yolov5_DeepSort_Pytorch部署 yolov5 pycharm_深度学习_13


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

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


Yolov5_DeepSort_Pytorch部署 yolov5 pycharm_虚拟环境_14


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

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


Yolov5_DeepSort_Pytorch部署 yolov5 pycharm_官网_15


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


Yolov5_DeepSort_Pytorch部署 yolov5 pycharm_深度学习_16


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


Yolov5_DeepSort_Pytorch部署 yolov5 pycharm_Powered by 金山文档_17


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


Yolov5_DeepSort_Pytorch部署 yolov5 pycharm_深度学习_18


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


Yolov5_DeepSort_Pytorch部署 yolov5 pycharm_虚拟环境_19


Yolov5_DeepSort_Pytorch部署 yolov5 pycharm_虚拟环境_20


四. 修改yolo参数

1、修改检测图像

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


Yolov5_DeepSort_Pytorch部署 yolov5 pycharm_深度学习_21


Yolov5_DeepSort_Pytorch部署 yolov5 pycharm_官网_22


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