第四节,目标检测---YOLO系列
1、R-CNN回顾
适应全卷积化CNN结构,提出全卷积化设计
- 共享ResNet的所有卷积层
- 引入变换敏感性(Translation variance)
- 位置敏感分值图(Position-sensitive score maps)
- 特殊设计的卷积层
- Grid位置信息+类别分值
- 位置敏感池化(Position-sensitive RoI pooling)
- 无训练参数
- 无全连接网络的类别推断
- 位置敏感分值图(Position-sensitive score maps)
R-FCN的位置敏感卷积层
使用k2(C+1)个通道对(位置,类别)组合进行编码
- 类别:C个物体类+1个背景类
- 相对位置:kxk个Grid(k=3)
- 位置敏感分值图(Position-sensitive score maps)
- 每个分类k2个score map
- score map 尺寸=图片尺寸
R-FCN的位置敏感RoI池化层
显式地编码相对位置信息
- 将wxh尺寸的RoI拆分成kxk个w/k x h/k尺寸的bin
- 不同(颜色)bin对应不同(颜色)通道层(score map)
- Bin内做均值池化
- 输出尺寸kxkx(C+1)
R-FCN的多任务损失函数
2、YOLO v1:GoogLeNet前20层
YOLO将物体检测任务当作一个regression问题来处理,通过YOLO,每张图像只需要“看一眼”就能得出图像中都有哪些物体和这些物体的位置。
- 将图像resize到448x448作为神经网络的输入,
- 使用一个神经网络,直接从一整张图像来预测出bbox的坐标、box中包含物体的置信度和物体的可能性,
- 然后进行非极大值抑制,筛选Boxes。
- 首先利用ImageNet 1000-class的分类任务数据集Pretrain卷积层,使用GoogLeNet中的前20个卷积层,加上一个average-pooling layer,最后加一个全连接层,作为Pretrain的网络。
- 将Pretrain的结果的前20层卷积层应用到Detection中,并加入剩下的4个卷积层及2个全连接层。
- 同时为了获取更精细化的结果,将输入图像的分辨率由224x224提升到448x448。
- 将所有的预测结果都归一化到0~1,使用Leaky RELU作为激活函数。
- 为了防止过拟合,在第一个全连接层后面接了一个ratio=0.5的Dropout层。
- YOLO网络结构由24个卷积层与2个全连接层构成,网络入口为448x448,图片进入网络先经过resize,网络的输出结果为一个张量。
- 输出维度:S*S*(B*5+C)
- 在Pascal VOC上,预测的结果维度是7*7*(2*5+20):其中S为划分网格数,B为每个网格负责目标个数,C为类别个数。
将一幅图像分成SxS个网格(grid cell),图中物体“狗”的中心点(红色原点)落入第5行、第2列的格子内,所以这个格子负责预测图像中的狗。
每一个栅格预测B(B=2)个bboxes,以及这些bboxes的confidence scores。
confidence scores反映了模型对于这个栅格的预测:该栅格是否含有物体,以及这个box的坐标预测的有多准。
如果这个栅格中不存在object,则confidence score应该为0;否则的话,confidence score则为predicted bound ing box与ground truth box之间的IoU(intersection over union)
- YOLO v1:每个栅格两个bbox:横向和纵向各有一个框,根据匹配情况选择合适的框;对每个bbox有5个预测值:x,y,w,h,confidence;作为预测的结果(生成的7*7*30的张量)每一行中前10个元素,后20个元素对应于20个类别的概率。
- 每一个栅格还要预测C条件类别概率(conditional class probability):Pr(Classi|Object)。即在一个栅格包含一个Object的前提下,它属于某个类的概率。
- 条件类别概率:是针对每个栅格的。
- Confidence:是针对每个bbox的。
- 在测试阶段,将每个栅格的条件类别概率与你每个bbox的confidence相乘:
NMS(非极大值抑制):
- 首先从所有的检测框中找到置信度最大的那个框;
- 然后挨个计算其余剩余框的IoU,如果其值大于一定阈值(重合度过高),那么就将该框剔除;
- 之后对剩余的检测框重复上述过程,直到处理完所有的检测框。
YOLO v1的优点:
- 检测物体的速度很快;
- 假阳性率低;
- 能够学到更加抽象的物体的特征
YOLO v1的不足:
- YOLO的物体检测精度低于其他state-of-the-art的物体检测系统;
- YOLO容易产生物体的定位错误;
- YOLO对小物体的检测效果不好(尤其是密集的小物体,因为一个栅格只能预测2个物体)。
3、YOLO v2改进
- Batch Normalization
- Batch Normalization可以提高模型收敛速度,减少过拟合,在所有卷积层都应用了BN,是结果提升了2%。
- High Resolution Classifier
- 基于ImageNet数据集预训练模型,大部分类器输入图像尺寸小于256x256;
- 在YOLO v2中,首先采用448*448分辨率的ImageNet数据fine tune使网络适应高分辨率 输入;
- 然后将该网络用于目标检测任务finetune。高分辨率输入使结果提升了4%mAP。
- Convolutional With Anchor Boxes
- 借鉴了Faster R-CNN中的anchor思想,用anchor boxes来预测bounding boxes。准确率只有小幅度的下降,而召回率则提升了7%。
- Dimension Clusters
- 使用了K-means聚类方法类训练bounding boxes,可以自动找到更好的boxes宽高维度。
- Multi-Scale Training
- 模型只包含卷积层和pooling层,因此可以随时改变输入尺寸。每经过10次训练,就会随机选择新的图片尺寸进行训练。
YOLO v2的基础模型是Darknrt-19:
使用较多的3*3卷积核,在每一次池化操作后把通道数翻倍。
网络使用了全局平均池化,把1*1的卷积核置于3*3的卷积核之间,用来压缩特征。
使用了BN稳定模型训练。
4、YOLO v3
YOLO v3中使用了一个53层的卷积网络,这个网络由残差单元叠加而成。
YOLO v3使用逻辑回归预测每个边界框的分数。
为了实现多标签分类,模型不再使用softmax函数作为最终的分类器,而是使用binary cross-entropy作为损失函数。
多尺度预测:YOLO v3从三种不同尺度夫人特征图谱上进行预测任务。
- 在Darknet-53得到的特征图的基础上,经过7个卷积得到第一个特征图谱,在这个特征图谱上做第一次预测。
- 然后从后向前获得倒数第3个卷积层的输出,进行一次卷积一次x2上采样,将上采样特征与第43个卷积特征连接,经过7个卷积得到第二个特征t图谱,在这个特征图谱上做第二次预测。
- 然后从后向前获得倒数第3个卷积层的输出,进行一次卷积一次x2上采样,将上采样特征与第26个卷积特征连接,经过7个卷积得到第三个特征图谱,在这个特征图谱上做第三次预测。
第四节,目标检测---YOLO系列的更多相关文章
- 小白也能弄得懂的目标检测YOLO系列之YOLOv1网络训练
上期给大家介绍了YOLO模型的检测系统和具体实现,YOLO是如何进行目标定位和目标分类的,这期主要给大家介绍YOLO是如何进行网络训练的,话不多说,马上开始! 前言: 输入图片首先被分成S*S个网格c ...
- 小白也能弄懂的目标检测YOLO系列之YOLOV1 - 第二期
上期给大家展示了用VisDrone数据集训练pytorch版YOLOV3模型的效果,介绍了什么是目标检测.目标检测目前比较流行的检测算法和效果比较以及YOLO的进化史,这期我们来讲解YOLO最原始V1 ...
- 深度学习笔记之目标检测算法系列(包括RCNN、Fast RCNN、Faster RCNN和SSD)
不多说,直接上干货! 本文一系列目标检测算法:RCNN, Fast RCNN, Faster RCNN代表当下目标检测的前沿水平,在github都给出了基于Caffe的源码. • RCNN RCN ...
- [目标检测]YOLO原理
1 YOLO 创新点: 端到端训练及推断 + 改革区域建议框式目标检测框架 + 实时目标检测 1.1 创新点 (1) 改革了区域建议框式检测框架: RCNN系列均需要生成建议框,在建议框上进行分类与回 ...
- 目标检测-yolo
论文下载:http://arxiv.org/abs/1506.02640 代码下载:https://github.com/pjreddie/darknet 1.创新点 端到端训练及推断 + 改革区域建 ...
- (四)目标检测算法之Fast R-CNN
系列博客链接: (一)目标检测概述 https://www.cnblogs.com/kongweisi/p/10894415.html (二)目标检测算法之R-CNN https://www.cnbl ...
- [目标检测]RCNN系列原理
1 RCNN 1.1 训练过程 (1) 训练时采用fine-tune方式: 先用Imagenet(1000类)训练,再用PASCAL VOC(21)类来fine-tune.使用这种方式训练能够提高8个 ...
- 目标检测YOLO进化史之yolov1
yolov3在目标检测领域可以算得上是state-of-art级别的了,在实时性和准确性上都有很好的保证.yolo也不是一开始就达到了这么好的效果,本身也是经历了不断地演进的. yolov1 测试图片 ...
- 目标检测YOLO算法-学习笔记
算法发展及对比: 17年底,mask-R CNN YOLO YOLO最大的优势就是快 原论文中流程,可以检测出20类物体. 红色网格-张量,在这样一个1×30的张量中保存的数据 横纵坐标中心点缩放到0 ...
随机推荐
- Oracle Metric sequence load elapsed time
Oracle Metric sequence load elapsed time The sequence load elapsed time Oracle metric is the amount ...
- 一位月薪1.2w的北漂程序员真实生活!
“ 每个人都有一条生活道路.千万人,千万条,各不相同,各有特点.但是并不是好人都有一条好路,坏人都有一坏路.有的时候却恰恰相反.这虽然不是历史的必然,但却是客观社会存在的.今天咱们说的这部书是发生在当 ...
- Spring Boot2.0使用Spring Security
一.Spring Secutity简介 Spring 是一个非常流行和成功的 Java 应用开发框架.Spring Security 基于 Spring 框架,提供了一套 Web 应用安全性 ...
- 几种c++字符串split 函数实现的比较
文中的字符串split函数功能是 从字符串中按照特定的分隔符进行分割,分割的结果保存到std::vector中. 1. strtok实现 std::vector<std::string> ...
- python和anacoda安装第三方库的位置
查看已安装库及版本号,命令行pip list 安装第三方库位置:
- Bean的自动装配
再说自动装配之前,我们先聊一聊什么是手动装配. 手动装配就是我们在先前讲的那些,要自己给定属性,然后赋值 Spring IOC容器可以自动装配Bean,需要做的仅仅实在<bean>的aut ...
- es6 模本字符串拼接方法 ``
1.字符串拼接 可以使用 es6 ` ` 配合 ${xxx} 具体操作上代码 <!DOCTYPE html> <html lang="en"> < ...
- java多线程中 volatile与synchronized的区别-阿里面试
volatile 与 synchronized 的比较(阿里面试官问的问题) ①volatile轻量级,只能修饰变量.synchronized重量级,还可修饰方法 ②volatile只能保证数据的可见 ...
- form表单中新增button按钮,点击按钮表单会进行提交
原生button控件,在非ie浏览器下,如果不指定type,默认为submit类型.如果不想自动提交表单,指定type=“button”
- npm后台启动nuxt服务之 kill
后台启动 npm run start & ps aux | grep start 根据项目对应的id执行如下命令 kill xxxx