基于PyTorch的YOLOv4目标检测

于 2024-04-28 14:04:58 发布
阅读量764 收藏 15
点赞数 11
文章标签： YOLO 目标检测 目标跟踪 深度学习 python pytorch 图像处理
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/wqn_wqn/article/details/138251448
版权

一、什么是YOLOv4

YOLOv4是一种高效且准确的目标检测算法，它是在YOLOv3的基础上进行了多项改进，从而实现了速度与精度的较大提升。具体来说，YOLOv4的贡献主要体现在以下几个方面：

  • 设计思路与框架原理：在输入端，YOLOv4使用了Mosaic数据增强、cmBN跨小批量标准化、SAT自对抗训练等技术；在BackBone层（主干网络），它采用了CSPDarknet53网络来提取特征，并结合了Mish激活函数和Dropblock正则化等策略。此外，YOLOv4还使用了多尺度训练和数据增强策略来提高模型的泛化能力和检测精度，以及多尺度推理和后处理技术来提高检测速度和准确性。
  • 性能表现：在MS COCO数据集上，YOLOv4实现了实时检测的最优表现，它在Tesla V100上以65 FPS运行，AP达到43.5%。与YOLOv3相比，YOLOv4的AP和FPS分别提高了10%和12%。
  • 应用场景：由于其高性能和广泛的应用前景，YOLOv4已被用于各种复杂场景的物体检测任务，如安防监控、自动驾驶、工业检测以及医疗影像分析等。

总的来说，YOLOv4是一种具有显著优势的目标检测算法，它不仅提升了检测速度和精度，而且具有广泛的适用性。如需更多关于YOLOv4的详细信息，建议查阅相关论文或技术文档。

本实验基于代码Github开源代码：GitHub - bubbliiiing/yolov4-pytorch: 这是一个YoloV4-pytorch的源码，可以用于训练自己的模型。

 二、环境配置

（一）Anaconda的安装

下载官网：Distribution | Anaconda

Anaconda安装的路径一定要是全英文的，有中文容易报错

使用Win+r 后输入 cmd

通过以下命令查看anaconda、python的版本

（二）Pytorch（深度学习框架）

PyTorch 是一个开源的Python机器学习库，PyTorch的前身是Torch，其底层和Torch框架一样，是一个以Python优先的深度学习框架，不仅能够实现强大的GPU加速I同时还支持动态神经网络。(即 PyTorch 既可以看作加入了GPU支持的numpy，同时也可以看成一个拥有自动求导功能的强大的深度神经网络。│

在Anaconda Prompt命令行依次输入以下四条命令切换到清华镜像源（先切换镜像源为国内清华镜像源，这样安装包的时候下载速度会快一些，也容易成功一些。）切换镜像源为非必要操作

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/

conda config --set show_channel_urls yes

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

查看镜像源命令：conda config --show channels，

切换到自己安装Anaconda目录的envs路径下,接下来创建一个名为pytorch的环境，并在环境中安装版本为3.8.5 的 python，命令如下(这里安装的 python版本根据自己的需要设置）。

conda create --name pytorch python=3.8

通过以下命令查看虚拟环境pytorch 是否已配置好python3.8.5版本

python --version

输入以下命令来查看已有的Conda环境信息，可以看到pytorch环境已经添加成功了

conda info -e

激活PyTorch环境：conda activate pytorch

1.安装pytorch

查看自己电脑cuda最大的支持版本（为pytorch安装做准备)

WIN+R进入命令框，输入cmd

在cmd中输入nvidia-smi ,(说明最大cuda支持版本为11.6)

到Pytorch官网复制适合自己电脑安装pytorch 的命令，如下图所示。

官网地址为: https://pytorch.org/

建议切换镜像源，即去掉末尾的默认pytorch 镜像，选择国内的清华大学镜像

conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=10.2 -c Index of /anaconda/cloud/pytorch/win-64/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror

将上面的命令输入到自己设置的 pytorch虚拟环境路径下

2.测试Pytorch深度学习框架是否成功安装

在pytorch环境下，输入python,进入python环境中输入import torch,顺利出现下一行即为成功

查看torch版本

import torch

print(torch.__version__)  #注意是双下划线

torch.cuda.is_available()

（三）pycharm的安装

官网下载地址：PyCharm: the Python IDE for data science and web development

  1. 访问官网下载安装包。打开PyCharm的官网，找到对应的版本进行下载。有两个版本可供选择，专业版和社区版。专业版功能更丰富，社区版则相对简化，对于初学者或日常学习使用，社区版通常就足够了。
  2. 安装PyCharm。下载完成后，双击安装包开始安装。首先选择安装路径，然后根据自己的需求选择相关配置，如是否创建桌面快捷方式、是否添加到系统环境变量等。之后点击“Install”进行安装，完成后可以选择立即重启或稍后重启。
  3. 激活与启动。安装完成后，首次启动PyCharm会提示用户许可证协议，需要同意后才能继续使用。如果是专业版，可能还需要进行激活操作。完成后，就可以开始使用PyCharm进行python开发了。

安装好后打开pycharm的setting，将anaconda的python解释器添加进来并使用。

在anaconda的Anaconda prompt（以管理员身份运行）中下载需要用到的python库（需先激活PyTorch环境并进入到你的envs文件夹下，再下载，这样不容易出错）

下载本次项目需要用到的python库：opencv-python、pillow、numpy、os、matplotlib、shutil、tqdm、datetime等（缺什么库就下载什么库即可）

在安装这些库时，我们可以使用镜像源来帮助我们快速安装：

南京邮电大学开源镜像网站地址：https://mirrors.njupt.edu.cn/

清华大学开源软件镜像站地址：https://mirror.tuna.tsinghua.edu.cn/

中国科大开源软件镜像站地址：http://mirrors4.ustc.edu.cn/

网易开源镜像站地址：http://mirrors.163.com/

阿里云开源镜像站地址：http://mirrors.aliyun.com/

腾讯云开源镜像站地址：https://mirrors.cloud.tencent.com/

豆瓣开源镜像站地址：https://pypi.douban.com/simple/

示例：pip install 下载的库名字 -i 开源镜像网址

pip install opencv-python -i https://pypi.douban.com/simple/

三、数据准备

（一）数据集的准备

准备我们想要检测的目标图像数据集，这里我想检测的是喜羊羊与灰太狼的人物识别，所以我爬取了二百多张关于喜羊羊与灰太狼的图片（数据集要图像500张以上更好），图片要统一格式，我统一的格式为jpg。准备完图像数据集需要使用labelImg_exe对jpg数据集进行标注成VOC数据集，不会标注的可以看相关博客：深度学习-使用Labelimg数据标注_深度学习图片标注 labelimg标注图片-CSDN博客

训练前将标签文件放在VOCdevkit文件夹下的VOC2007文件夹下的Annotation中。

训练前将图片文件放在VOCdevkit文件夹下的VOC2007文件夹下的JPEGImages中。


训练所需的数据集可在百度网盘中下载。VOC数据集下载地址如下，里面已经包括了训练集、测试集、验证集（与测试集一样），无需再次划分：

链接: https://pan.baidu.com/s/19Mw2u_df_nBzsC2lg20fQA   提取码: j5ge

 （二）设置检测目标标签

训练自己的数据集时，可以自己建立一个cls_classes.txt，里面写自己所需要区分的类别。或者在model_data下的voc_classes.txt里添加目标标签，要全英文的，不能含有中文和其他字符。

（三）数据集的处理

在完成数据集的摆放之后，我们需要利用voc_annotation.py获得训练用的2007_train.txt和2007_val.txt。 修改voc_annotation.py里面的annotation_mode=2，第一次训练可以仅修改classes_path，classes_path用于指向检测类别所对应的txt。 运行voc_annotation.py生成根目录下的2007_train.txt和2007_val.txt。

（四）检查路径

仔细检查train.py里的模型路径，标签路径，数据集路径等是否正确，不然容易报错找不到文件。

路径文件夹不能为空，且不可以存在空格，一定要仔细检查，不然容易报错。

四、训练过程

训练所需的yolo4_weights.pth可在百度网盘中下载。

链接: https://pan.baidu.com/s/1oXz13QwLx1lnXct538qL2Q 提取码: 16qc

yolo4_weights.pth是coco数据集的权重。

yolo4_voc_weights.pth是voc数据集的权重。

train.py的默认参数用于训练VOC数据集，直接运行train.py即可开始训练。（需要仔细阅读注释，第一次训练大部分参数不需要修改，只需要修改对应的classes_path路径，使路径正确）

训练成功如下，输出我们设置的参数，还有训练成功的进度条：

五、训练结果预测

训练好模型，我们就可以使用predict.py文件进行预测结果，不需要修改，直接运行即可。

运行输入预测图片的路径，确认后它会运行弹出预测图片（预测会有线框），由于我的数据集少，训练效果很不好，预测不出来，要么没有线框，要么准确率很低。为什么我的训练效果不好？预测了没有框（框不准）我在网上找到了一些解决办法1、目标信息问题，查看2007_train.txt文件是否有目标信息，没有的话请修改voc_annotation.py。 2、数据集问题，小于500的自行考虑增加数据集，同时测试不同的模型，确认数据集是好的。 3、是否解冻训练，如果数据集分布与常规画面差距过大需要进一步解冻训练，调整主干，加强特征提取能力。 4、网络问题，比如SSD不适合小目标，因为先验框固定了。 5、训练时长问题，只训练了几代表示没有效果，按默认参数训练完。 6、确认自己是否按照步骤去做了，如果比如voc_annotation.py里面的classes是否修改了等。 7、不同网络的LOSS不同，LOSS只是一个参考指标，用于查看网络是否收敛，而非评价网络好坏，LOSS的值不重要，重要的是是否收敛。 8、是否修改了网络的主干，如果修改了没有预训练权重，网络不容易收敛，自然效果不好。

所以很有可能是我的数据集不够，我需要增加数据集。

预测其他数据集准确有线框图片示例:

YOLOv4中的get_map文件是用于计算模型在目标检测任务上的平均准确率均值（mAP，mean Average Precision）的工具。mAP是衡量目标检测算法性能的重要指标，它综合考虑了模型在不同物体类别上的检测性能。

预测准确可以使用get_map.py文件计算性能指标，先打开yolo.py文件设置model_path地址填入logs下训练好的loss模型，例如我下图的ep010-loss0.263-val_loss0.206.pth。classes_path与训练和预测的一致即可。

然后将get_map里的classes_path里与训练和预测的一致即可，map_mode根据需求修改，默认为0。

运行get_map文件后，它会加载测试集图像和对应的标注文件，然后利用训练好的模型对测试集进行推理。对于每个测试图像，模型会输出预测的目标物体类别和边界框坐标。

get_map文件会根据模型的预测结果和标注文件中的真实标签计算每个类别的平均准确率（AP，Average Precision）。然后，它会将所有类别的AP值进行平均，得到最终的mAP值。

在计算AP值时，通常会使用IoU（Intersection over Union）阈值来判断预测边界框与真实边界框的匹配程度。IoU阈值是一个介于0到1之间的数值，用于确定预测是否正确。例如，如果IoU阈值设置为0.5，那么只有当预测边界框与真实边界框的IoU值大于或等于0.5时，才认为该预测是正确的。

由于我的get_map有问题识别不出来定义函数，get_map运行不出来，这里就不做示例。

六、总结

本文介绍了使用YOLOv4算法进行喜羊羊与灰太狼动画人物目标检测的过程。通过环境搭建、准备数据集、训练模型、测试模型等步骤，展示了YOLOv4在动画人物检测方面的应用。虽然在实际应用中可能还存在一些问题，但YOLOv4算法的高效性和准确性使其在目标检测领域具有广泛的应用前景。

在这次作业中我遇到许多的问题，我也在网上寻找了一些解决办法，这使我学习到很多，例如调参和目标检测的原理等

想要训练自己的数据集，数据集的准备是目标检测任务中至关重要的一步，需要收集了大量的喜羊羊与灰太狼动画图片，并进行了精确的标注工作。这个过程虽然繁琐，但对于确保模型的训练效果是不可或缺的。一个完整、准确的数据集能够提供足够的信息供模型学习，从而使其能够更好地识别目标物体。通过这次作业，我也深刻体会到了深度学习的强大和魅力。使用官方的数据集进行训练，YOLOv4算法的高效性和准确性让我对目标检测任务有了更深入的了解。同时，我也意识到数据集的准备和模型的训练是目标检测任务中不可或缺的两个环节。只有在这两个环节上做好充分的准备和优化，才能取得更好的检测效果。我相信随着深度学习技术的不断发展和完善，目标检测算法的性能将会得到进一步的提升。同时，我们也可以将目标检测技术应用于更多的领域和场景中，如人脸识别、物体追踪等，为实际应用提供更多可能性。

注：本人新手小白，第一次使用YOLOv4模型，如有错误望指正，可以私信或评论区留下您的见解。

發财#
关注
  • 11
    点赞
  • 踩
  • 15
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
yolov4 训练自己的数据集
reset2021的博客
07-22 1277
基于yolov4算法训练自己的数据集，从实操入手
YOLOv4目标检测
01-03
YOLOv4目标检测,介绍YOLOv4使用方法，训练
Python实现yolov4(tiny)（摄像头实现）
略知12的博客
04-06 1920
环境要求：基本的Pytorch及相关模块 模型文件：见https://blog.csdn.net/qq_37249793/article/details/115467880 代码： # -*- coding: utf-8 -*- """ Created on Sat Mar 13 10:24:50 2021 @author: S """ import numpy as np import cv2 import os import time def video_demo(): # 加载
目标检测——Yolov4
zyw2002的博客
02-17 1894
文章目录Bag of freebies(BOF)数据增强网络正则化的方法类别不平衡，损失函数设计Bag of specials(BOS)SPPNet(Spatial Pyramid Pooling)CSPNet（Cross Stage Partial Network）CBAM（Convolutional Block Attention Module）PAN(Path Aggregation Network)激活函数Misheliminate grid sensitivity整体网络架构 Bag of fre
YOLOv4目标检测-Backbone
一个努力写代码的小菜鸟，目前主要从事图像算法，目标检测/分割，模型量化与部署的工作
02-12 3357
YOLOv4源码详解笔记---Backbone---CSPDarknet53
目标检测之YOLOv4
qq_41917697的博客
01-01 2399
目标检测之YOLOv4框架技巧Trick/优化1.BoF2.BoSYOLO V4 论文地址：YOLOv4: Optimal Speed and Accuracy of Object Detection 源代码：https://github.com/AlexeyAB/darknet，原作者YOLO V4的代码是基于C++的 pytorch实现： https://github.com/GZQ0723/YoloV4 https://github.com/Tianxiaomo/pytorch-YOLOv4 YOLO
基于pytorch的yolov5目标检测框架
02-26
**基于PyTorch的YOLOv5目标检测框架详解** YOLO（You Only Look Once）是一种实时目标检测系统，自2016年首次提出以来，已经发展到多个版本，其中YOLOv5是最新且性能优越的一个。该框架以其高效、准确的目标检测...
yolov5s目标检测模型，基于pytorch实现
04-15
总的来说，YOLOv5s是基于PyTorch的目标检测模型，以其高效和准确的特性在实际应用中受到欢迎。通过理解模型的工作原理，掌握深度学习和PyTorch的基本知识，开发者可以有效地利用YOLOv5s解决各种目标检测问题，例如...
基于pytorch实现yolov5目标检测，返回检测目标相机坐标系下的位置信息
最新发布
05-22
【作品名称】：使用realsense d435i相机，基于pytorch实现yolov5目标检测，返回检测目标相机坐标系下的位置信息 【适用人群】：适用于希望学习不同技术领域的小白或进阶学习者。可作为毕设项目、课程设计、大作业、...
基于pytorch实现yolov5目标检测返回检测目标相机坐标系下的位置
01-30
使用realsense d435i相机，基于pytorch实现yolov5目标检测，实时返回检测目标相机坐标系下的位置信息。 1.Environment： 1.一个可以运行YOLOv5的python环境 pip install -r requirements.txt 2.一个realsense相机...
目标检测最强算法YOLOv4 源码（source：GitHub）
05-15
如果Github下载较慢的话，可以使用这个文件 ，原github地址： ghttps://github.com/AlexeyAB/darknet
Yolov4：这是一个yolov4_pytorch代码
03-03
约洛夫4 这是一个yolov4_pytorch代码 没有实现所有trick，做了以下实现： 骨干：DarkNet53 => CSPDarkNet53 特征金字塔：SPP，PAN 训练：Mosaic数据增强，标签平滑，学习率余弦衰减衰减，CIOU 激活函数：Mish
yolov4_python
03-19
yolov4_python
WIN10+YOLOv4目标检测.pdf
08-18
WIN10+YOLOv4，windows上完美执行YOLOv4目标检测 1 软件安装 2 配置YOLOv4环境 3 目标检测
PyTorch_YOLOv4:YOLOv4的PyTorch实现
03-18
YOLOv4 这是PyTorch实施这是基于 。 。 开发日志 扩张 2021-01-26设计蒙版的2021-01-26 。 2021-01-25设计旋转增强。 2021-01-23设计拼贴增强。 2021-01-22支持 ， 。 2021-01-22支持 。 2021-01-19支持实例细分。 2021-01-17支持基于免锚的方法。 2021-01-14支持联合检测和分类。 2020-01-02年1 2020-01-02 -设计基于和的新模型。 2020-12-22支持转学。 2020-12-18支持非本地系列自我关注块 2020-12-16在cspnet纸中支持下采样块。 2020-12-03支持模仿学习 2020-12-02支撑。 2020-11-26支持多类多锚联合检测和嵌入。 2020-11-25支持。 2020-11-23支持师生学习。
基于python的YoloV4-pytorch目标检测模型实现代码
03-24
**基于Python的YoloV4-pytorch目标检测模型实现** 在深度学习领域，目标检测是一种重要的计算机视觉任务，它旨在识别图像中的特定对象并精确地框出它们的位置。Yolo（You Only Look Once）是一种高效的目标检测算法...
目标检测算法(YOLOv4)
Victory_gx的博客
02-20 5133
目标检测算法(yolov4) 本文要来介绍很火的目标检测算法YOLOv4，该算法是在原有YOLO目标检测架构的基础上，采用了近些年CNN领域中最优秀的优化策略，从数据处理、主干网络、网络训练、激活函数、损失函数等各个方面都有着不同程度的优化. 论文题目：YOLOv4: Optimal Speed and Accuracy of Object Detection 主要贡献 YOLOv4在YOLOv3上各个部分做改进,在保证速度的同时,大幅提高了模型的检测精度,并降低硬件使用的要求. 由上图可以看出: Y
【目标检测】PyTorch版YOLOv4项目介绍
qq_25692343的博客
10-05 1227
文章原文出处 https://mp.weixin.qq.com/s/3ItH0swoicI_gMmfkHHClQ. 背景知识 YOLO v4框架采用C语言作为底层代码。 近日，有研究者在 GitHub 上开源了一个项目：基于 PyTorch 深度学习框架的 YOLOv4 复现版本，该版本基于 YOLOv4 作者给出的实现 AlexeyAB/darknet，并在 PASCAL VOC、COCO 和自定义数据集上运行。 项目介绍 项目地址: https://github.com/argusswift/YOLOv
yolo v4 python_Yolov4-训练自己的数据集（非ROS）
weixin_39651816的博客
12-09 626
一、前言上一期分享了如何安装和配置Yolov4，还有安装编译Opencv-3.4.10，Cuda等，若没安装的欢迎点击下面这篇进行参考。本文主要介绍如何基于自己的数据集进行训练，并获取自己的权重因子，用于自己的项目，只要有自己标注的数据集，那么基本检测什么都行，注意这里最好的迁移结果数据集放到2000张左右，如果没有那么多，100多张也勉强够用。追逐雅克比：Yolov4配置-Ubuntu18.04...
【目标检测-YOLO】YOLOv4（第一篇）
博观约取，厚积薄发
03-05 2983
YOLO v4 论文翻译
评论 2
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值