目标检测入门系列手册五:YOLO训练教程

简介: 目标检测入门系列手册五:YOLO训练教程

>>戳此处立即下载电子书<<,学习全套目标检测算法&模型

image.png

YOLO

由于在R-CNN 的系列算法中都需要首先获取大量proposal,但proposal 之间有很大的重叠,会带来很多重复的工作。YOLO[5] 一改基于proposal 的预测思路,将输入图片划分成SxS 个小格子,在每个小格子中做预测,最终将结果合并,如图2-14 所示。接下来我们看一下YOLO 学习的关键步骤:

(1)YOLO 对于网络输入图片的尺寸有要求,首先需要将图片缩放到指定尺寸(448x448),再将图片划分成SxS 的小格。

(2)每个小格里面做这几个预测:该小格是否包含物体、包含物体对应的矩形框位置以及该小格对应C 个类别的分数是多少。因此,每个小格需要预测的的维度为B x(1+4)+ C,其中B 代表每个小格最多可能交叠物体的个数,1 为该小格是否包含物体的置信度,4 用来预测矩形框,C 表示任务中所有可能的类别个数(不包含背景)。因此,YOLO 网络最终特征层的大小为 S x S x( Bx5 + C),图 2-14 中特征
层大小即为 7 x 7 x ( 2 x 5 + 20)=7x7x30(Pascal VOC2012 目标检测数据集共有20 种类别)。

由于YOLO 直接将输入图片划分为SxS 个小格,不需要产生proposal 的过程,所以速度比Faster R-CNN 快很多,但是因为粒度较粗,所以精度相比Faster R-CNN 略逊一筹。YOLO 的主要贡献是为目标检测提供了另一种思路,并使实时目标检测成为可能。近几年,YOLOv2 和YOLOv3 接连推出,感兴趣的读者可以参考附录的6。

image.png

图2-14 基于Pascal VOC2012 目标检测数据集的YOLO 示意图

image.png

目录
相关文章
|
编解码 缓存 并行计算
YOLOv5入门实践(4)——手把手教你训练自己的数据集
YOLOv5入门实践(4)——手把手教你训练自己的数据集
2288 0
YOLOv5入门实践(4)——手把手教你训练自己的数据集
|
2月前
|
数据处理 算法框架/工具 计算机视觉
手把手教你使用YOLOV5训练自己的目标检测模型
本教程由肆十二(dejahu)撰写,详细介绍了如何使用YOLOV5训练口罩检测模型,涵盖环境配置、数据标注、模型训练、评估与使用等环节,适合大作业及毕业设计参考。提供B站视频、CSDN博客及代码资源链接,便于学习实践。
167 1
手把手教你使用YOLOV5训练自己的目标检测模型
|
7月前
|
机器学习/深度学习 算法 PyTorch
YOLO如何入门?
YOLO如何入门?
|
6月前
|
机器学习/深度学习 数据可视化 TensorFlow
【手把手教学】如何可视化YOLOv8深度学习的网络结构并保存
【手把手教学】如何可视化YOLOv8深度学习的网络结构并保存
|
7月前
|
机器学习/深度学习 PyTorch 测试技术
PyTorch实战:图像分类任务的实现与优化
【4月更文挑战第17天】本文介绍了使用PyTorch实现图像分类任务的步骤,包括数据集准备(如使用CIFAR-10数据集)、构建简单的CNN模型、训练与优化模型以及测试模型性能。在训练过程中,使用了交叉熵损失和SGD优化器。此外,文章还讨论了提升模型性能的策略,如调整模型结构、数据增强、正则化和利用预训练模型。通过本文,读者可掌握基础的PyTorch图像分类实践。
|
7月前
|
机器学习/深度学习 数据采集 PyTorch
PyTorch搭建卷积神经网络(ResNet-50网络)进行图像分类实战(附源码和数据集)
PyTorch搭建卷积神经网络(ResNet-50网络)进行图像分类实战(附源码和数据集)
269 1
|
XML 计算机视觉 数据格式
YOLOv5入门实践(2)——手把手教你利用labelimg标注数据集
YOLOv5入门实践(2)——手把手教你利用labelimg标注数据集
6999 1
YOLOv5入门实践(2)——手把手教你利用labelimg标注数据集
|
机器学习/深度学习 人工智能 计算机视觉
【YOLOv8】实战一:手把手教你使用YOLOv8实现实时目标检测
【YOLOv8】实战一:手把手教你使用YOLOv8实现实时目标检测
18820 0
【YOLOv8】实战一:手把手教你使用YOLOv8实现实时目标检测
|
机器学习/深度学习 数据可视化 算法
【Pytorch神经网络实战案例】23 使用ImagNet的预训练模型识别图片内容
Imagenet数据集是目前深度学习图像领域应用得非常多的一个领域,关于图像分类、定位、检测等研究工作大多基于此数据集展开。
677 0
|
并行计算 PyTorch 算法框架/工具