机器学习笔记 - 基于Torch Hub和YOLOv5和SSD的目标检测

一、关于对象检测

         对象检测是一个很有意思的领域。让机器识别图像中物体的确切位置有各种各样的关键用途。从人脸检测系统到帮助自动驾驶汽车安全导航，不胜枚举。

1、YOLO

        YOLO 或“ You Only Look Once ”（2015 年）引入了一种巧妙的方法来解决简单CNN检测器的缺点。我们将每个图像分割成一个S × S网格，得到每个单元格对应的对象位置。当然，有些单元格不会有任何对象，而有些单元格会出现在多个单元格中，见下图。

         了解完整图像的对象的中点、高度和宽度非常重要。然后每个单元格将输出一个概率值（对象在单元格中的概率）、检测到的对象类别以及该单元格唯一的边界框值。即使每个单元格只能检测到一个对象，多个单元格的存在也会使约束无效。结果如下图。