【模型复现】高精度端到端目标检测模型—CenterNET复现

【模型复现】高精度端到端目标检测模型—CenterNET复现

极链AI云 注册链接
关注极链AI云公众号，学习更多知识！

---

一、模型详情

1.1 模型简介：

图像检测中的检测框默认是轴对称的，大多数比较成功的检测方法都是要列举很大数量的候选框，再进一步进行定位和分类。但是这样的做法往往是多余而又低效率的。本文中我们提出了一种对物体的中心点进行定位的方法，并且在此基础上回归出物体的其他属性，比如尺寸、3D位置、方向甚至是姿态。我们提出的网络称之为CenterNet，它是一个端到端可微分，更简化快捷精度更高的方法。

1.2 关键词：

高精度端到端

1.3 应用场景：

目标检测

1.4 结构：

1.5 论文地址：

https://arxiv.org/abs/1904.08189v3

1.6 模型来源：

https://github.com/bubbliiiing/centernet-pytorch

---

下面就是该模型的具体复现流程

二、平台环境准备

2.