Nanodet:手机端的 97FPS 的 Anchor-free 的目标检测模型

参考资料:

YOLO之外的另一选择,手机端97FPS的Anchor-Free目标检测模型NanoDet现已开源~

1、什么是 Nanodet

Nanodet 是一个速度超快和轻量级的移动端 Anchor-free 目标检测模型,是基于 FCOS 模型进行轻量化改进而来的


2、Nanodet 跟其他模型的性能对比


华为 P30 上用 NCNN 移植跑 benchmark,一张图片仅需 10.23 毫秒,比 YoloV4-Tiny 快 3 倍,参数量小 6 倍,COCO mAP(05:0.95) 能够达到 20.6。而且模型权重文件仅有 1.8m

由此可见,Nanodet 能够在有较低参数量和较小的模型权重文件情况下,能够拥有跟 YoloV4-Tiny 一样的 mAP


3、Nanodet 的网络结构图


backbone 是 ShuffNetV2 1.0x,去掉最后一层的卷积层,并且抽取 8/16/32倍下采样率的特征输入到 PAN 中做多尺度的特征融合


超轻量的 PAN


4、Nanodet 相对于 FCOS 的改进


FCOS 模型结构

① 检测头

FCOS:共享检测头权重

Nanodet:不共享检测头,因为在移动端是使用 CPU 来进行计算的,共享检测头权重并不会对推理过程加速;共享检测头会使得其检测能力进一步下降

② 归一化方式

FCOS:使用 Group Normalization

Nanodet:使用 Batch Normalization 来取代 Group Normalization,因为后者虽然比前者有很多优点,但却有一个缺点:Batch Normalization 可以在推理时将其参数融入到卷积核中,节省这一步计算可以节省时间

③ 检测头大小

FCOS:检测头大小为 4 个 256 通道数的卷积核组为一个分支,因此边框回归和分类两个分支,则总共需要 8 个 256 通道数的卷积

Nanodet:将 4 个卷积核组减少为 2 个卷积核组。在通道数上将 256 维降低为 96 维。将边框回归和分类共享检测头,然后再切割成两份,分别去预测边框回归和分类。

④ FPN 层

FCOS:没有该模块

Nanodet:基于 FPN 改进为 PAN,原版的 FPN 在对高层特征图使用线性插值后,再使用 3*3 卷积。但是 Nanodet 去掉了 FPN 中线性插值后再使用的 3*3 卷积,只保留从骨干网络特征提取后的 1*1 卷积来进行特征通道维度的对齐。同时 FPN 只是在多层特征融合端使用了自上而下的特征融合,而 PAN 则不仅使用了自上而下的特征融合,而且还使用了自底向上的特征融合,使用的方法也是线性插值。


5、Nanodet 的算法步骤



6、Nanodet 的损失函数


7、Nanodet 的优点

① 速度快

② 模型参数权重文件小


8、Nanodet 的缺点

① mAP 不高

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 203,098评论 5 476
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,213评论 2 380
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 149,960评论 0 336
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,519评论 1 273
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,512评论 5 364
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,533评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,914评论 3 395
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,574评论 0 256
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,804评论 1 296
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,563评论 2 319
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,644评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,350评论 4 318
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,933评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,908评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,146评论 1 259
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 42,847评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,361评论 2 342

推荐阅读更多精彩内容