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

本文详细指导如何在Pycharm中从YOLOv5GitHub项目下载代码,安装所需的库文件(包括PyTorch、torchvision等),以及检查GPU可用性,确保目标检测脚本detect.py的成功运行。
摘要由CSDN通过智能技术生成

前提是下载好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
2.1.1安装

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

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

如果输出为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

2.3 thop库的安装

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

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

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

在这里插入图片描述

3.2原来的数据文件

在这里插入图片描述

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

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

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

  • 2
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
基于Yolov5的雾天行人车辆检测可以通过以下步骤实现: 1.准备数据集:收集并标注雾天环境下的行人和车辆图像数据集。 2.训练模型:使用Yolov5算法对数据集进行训练,得到行人和车辆目标检测模型。 3.模型测试:使用训练好的模型对新的雾天环境下的图像进行目标检测。 具体实现步骤如下: 1.安装Yolov5:在终端中输入以下命令安装Yolov5。 ```shell git clone https://github.com/ultralytics/yolov5.git cd yolov5 pip install -r requirements.txt ``` 2.准备数据集:收集并标注雾天环境下的行人和车辆图像数据集。将数据集划分为训练集和测试集,并将其放置在相应的文件夹中。 3.训练模型:在终端中输入以下命令训练模型。 ```shell python train.py --img 640 --batch 16 --epochs 100 --data data.yaml --cfg models/yolov5s.yaml --weights '' --name yolov5s_results ``` 其中,--img指定输入图像的大小,--batch指定批量大小,--epochs指定训练轮数,--data指定数据集的配置文件,--cfg指定模型的配置文件,--weights指定预训练模型的权重文件,--name指定训练结果的保存文件夹。 4.模型测试:在终端中输入以下命令测试模型。 ```shell python detect.py --weights runs/train/yolov5s_results/weights/best.pt --img 640 --conf 0.4 --source test/images/ ``` 其中,--weights指定训练好的模型的权重文件,--img指定输入图像的大小,--conf指定置信度阈值,--source指定测试集的图像文件夹。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值