迈微视觉 | 用CenterNet对旋转目标进行检测

关注“迈微AI研习社”，内容首发于公众号

作者丨不是大叔@知乎

来源丨https://zhuanlan.zhihu.com/p/163696749

编辑丨迈微AI研习社

 

迈微导读

作者对CenterNet进行了部分修改，以此进行旋转目标检测。本文介绍了这种工作的代码、数据处理和样例，供读者参考。

 

前言

前段时间纯粹为了论文凑字数做的一个工作，本文不对CenterNet原理进行详细解读，如果你对CenterNet原理不了解，建议简单读一下原论文然后对照本文代码理解（对原版CenterNet目标检测代码进行了极大程度精简）。

基本想法就是直接修改CenterNet的head部分，但是是直接在长宽图上加一层通道表示角度，还是多引出一 路feature-map呢？实测是多引出一张feature map比较好，直接在长宽图上加一层通道很难收敛。具体原因我也是猜测，