1 前言

CenterNet的代码还是有点难懂，不过还是要感谢各位同学分享的资料～

2 CenterNet代码的学习笔记

2.1 数据读取——COCO类

COCO类是用来进行数据读取的，读取之后获得的item有以下几个属性：

'image': img,
'hmap': hmap,
'w_h_': w_h_,
'regs': regs,
'inds': inds——
inds指的是目标中心的线式索引，其中“线式索引”指的是将单张特征图平铺为一维向量之后的索引（或者说坐标），
从inds的计算公式也可以看出：inds[k] = obj_c_int[1] * self.fmap_size['w'] + obj_c_int[0]'ind_masks': ind_masks,'c': center, 's': scale, 'img_id': img_id

2.2 后处理

下面在介绍后处理时我们会按照测试时的顺序进行介绍；

2.2.1 _nms——基于高斯分布特性的NMS算法

使用检测局部极大值的方法，来获得检测的峰值，从而获得得分最高的检测框；

如图所示：

2.2.2 Top-K——获取得分最高的k个值

使用Top-K方法来获取得分最高的k个值，k是一个超参数；

如图所示：

在论文中，K=100；

2.2.3 Soft-NMS——在测试阶段对boxes进行过滤

CenterNet在测试阶段时使用了Soft-NMS算法，

我觉得可能的原因是CenterNet在测试时将图片建立了多尺度输入，所以很可能会出现重复检测框的情况，

2.3.4 max_per_image——实际上用的也是Top-K进行筛选

通过设定每种图片最多的目标数量来进筛选，通过max_per_image来获取进行筛选的阈值；

如图所示：

自动驾驶——CenterNet（Objects as Points）的学习笔记相关推荐

CenterNet:Objects as Points论文阅读笔记
CenterNet论文阅读笔记 (一)Title (二)Summary (三)Research Objective (四)Problem Statement (五)Method 5.1 Loss Fu ...
CenterNet：Objects as Points论文学习笔记+代码复现(demo+训练数据)【检测部分】
目录 1.关键部分Heatmap了解 2.Centernet论文细节: 3.尝试复现CneterNet--INSTALL.md安装: 4.尝试复现CneterNet--跑跑demo.py: 5.尝试复 ...
论文精读——CenterNet :Objects as Points
论文精读--CenterNet :Objects as Points 论文地址:https://arxiv.org/pdf/1904.07850.pdf 机构:UT Austin,UC Berkele ...
centernet: objects as points
轻松掌握 MMDetection 中常用算法(七):CenterNet - 知乎文@ 0000070 摘要在大家的千呼万唤中,MMDetection 支持 CenterNet 了!! CenterN ...
CenterNet :Objects as Points 详解
论文题目:Objects as Points ...
使用 Carla 和 Python 的自动驾驶汽车第 4 部分 —— 强化学习Action
欢迎来到自动驾驶汽车的第五部分,并与Carla.Python和TensorFlow加强学习. 现在我们已经有了环境和代理,我们只需要添加更多的逻辑将它们连接在一起,这是我们接下来要做的. 首先,我们将 ...
使用 Carla 和 Python 的自动驾驶汽车第 4 部分 —— 强化学习代理
在我们的自动驾驶汽车的第四部分,Carla, Python, TensorFlow,和强化学习项目,我们将为我们的实际代理编码.在前一篇教程中,我们研究了环境类,我们的代理将与之交互. 在考虑如何创建 ...
看代码解读CenterNet :Objects as Points
文章目录 1. idea 2. 网络结果 2.1 hm分支 2.2 wh分支 2.3 reg分支 3. 数据+loss 4. 推理结果 reference 摘要: centernet是单阶段的目标检 ...
自动驾驶系统设计及应用--清华大学出版--部分笔记
作者:余贵珍.周彬.王阳.周亦威.白宇目录第一章自动驾驶系统概述 1.1 自动驾驶系统架构 1.1.1 自动驾驶系统的三个层级 1.1.2 自动驾驶系统的基本技术架构 1.2 自动驾驶技术国内外 ...

自动驾驶——CenterNet（Objects as Points）的学习笔记

1 前言

2 CenterNet代码的学习笔记

2.1 数据读取——COCO类

2.2 后处理

2.2.1 _nms——基于高斯分布特性的NMS算法

2.2.2 Top-K——获取得分最高的k个值

2.2.3 Soft-NMS——在测试阶段对boxes进行过滤

2.3.4 max_per_image——实际上用的也是Top-K进行筛选

自动驾驶——CenterNet（Objects as Points）的学习笔记相关推荐

最新文章

热门文章