当前位置:   article > 正文

基于yolov5的车辆目标检测(一)yolov5环境的配置_gitpython>=3.1.30

gitpython>=3.1.30

前提是下载好Pycharm和Anaconda

在开始之前,我们首先去YOLOv5的官方项目里下载代码:
YOLOV5GitHub下载地址
点击下载
点击下载到自定义文件夹

1.在Pycharm中打开下载好的文件YOLOv5

在这里插入图片描述
如图所示,打开文件中的requirement,txt 安装对应的库文件

2.安装对应的库文件

2.1Pytorch文件安装

打开cmd,首先创建虚拟环境,分别输入下面两条命令,完成虚拟环境的创建和激活

1 conda creat -n yolov5  python==3.8
2 conda activate yolov5
  • 1
  • 2
2.1.1安装

pytorch下载网站
博主选择的10.1的版本,选择10.2的会安装失败

2.1.2测试GPU是否可用
1 import torch
2 print(torch.cuda.is_available())
  • 1
  • 2

如果输出为True则表示GPU可以使用
在这里插入图片描述

2.2 其余库的安装

pip install 库名字

gitpython>=3.1.30
matplotlib>=3.3
numpy>=1.22.2
opencv-python>=4.1.1
Pillow>=7.1.2
psutil  # system resources
PyYAML>=5.3.1
requests>=2.23.0
scipy>=1.4.1
thop>=0.1.1  # FLOPs computation
torch>=1.8.0  # see https://pytorch.org/get-started/locally (recommended)
torchvision>=0.9.0
tqdm>=4.64.0
ultralytics>=8.0.147
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14

2.3 thop库的安装

# 然后在虚拟环境下安装
pip install thop -i https://pypi.tuna.tsinghua.edu.cn/simple somepackages
  • 1
  • 2

3.运行查看是否安装成功

3.1运行detect.py脚本,安装成功之后,会出现命令行如情况

在这里插入图片描述

3.2原来的数据文件

在这里插入图片描述

3.3目标检测成功后的现象:

在这里插入图片描述
在这里插入图片描述

经过不断尝试,试错之后终于安装成功,希望可以解决大家遇到的问题。生活不易,大家浏览请点赞并关注。

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

闽ICP备14008679号