使用Yolov4实现简单的图片标注

一、前言
YOLOv4（You Only Look Once version 4）是一种先进的目标检测系统，于2020年推出。它是对之前版本YOLO的改进，后者是一种广泛使用的基于深度学习的目标检测算法。YOLOv4基于深度卷积神经网络，可以高精度实时检测图像中的目标。它使用单个神经网络，输入图像并输出所有目标的边界框和类别概率。YOLOv4相对于其前身YOLOv3的一些主要改进包括：

• 增加网络的深度和宽度，以获得更好的特征表示
• 使用先进的数据增强技术，以提高模型的泛化能力
• 整合多种先进的目标检测技术，如空间金字塔池化、Mish激活函数和交叉阶段部分网络。

YOLOv4是一个开源项目，在GitHub上免费提供。它已成为许多计算机视觉应用程序的流行选择，包括自动驾驶汽车、安全系统和监控系统。论文 ，代码。

本实验基于代码Github开源代码：

项目首页 - yolov4-pytorch - GitCodehttps://gitcode.com/bubbliiiing/yolov4-pytorch/overview?utm_source=csdn_github_accelerator&isLogin=1

 二、运行环境准备

运行文件前，需要的主要的库（依赖）我们都可以在requirements.txt中找到。

我们直接运行下面这个代码就可以安装所需要的库（如果下载太慢可以在后面加一个源，只要是国内的，什么源都可以）。如果下面代码运行不成功，那只能辛苦一下自己，一个一个下载。总之，没有什么库就下什么库。

pip install -r requirements.txt

pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple/

在这里如果条件允许的话，下载torch的时候选择cuda版本，这样训练模型的时候会快一点。因为我的是集显的笔记本，而且安装cuda版本的话很麻烦，所以我的是cpu版本的torch，训练的时候真的是很久。

如何正确下载PyTorch、CUDA版本！！_cuda12.0下载-优快云博客