arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 1712.00960cs.CV

FSSD:特征融合单次多框检测器

FSSD: Feature Fusion Single Shot Multibox Detector

Beihang University(北京航空航天大学);Beijing University of Posts and Telecommunications(北京邮电大学)

Zuoxin Li, Lu Yang, Fuqiang Zhou

AI总结:

本文提出 FSSD,通过在 SSD 中引入轻量级多尺度特征融合模块生成新特征金字塔,在 Pascal VOC 和 COCO 上以极小的速度代价显著提升检测精度,同时兼顾速度与精度。

摘要:

SSD(单次多框检测器)是兼具高精度与高速度的最佳目标检测算法之一。然而,SSD 的特征金字塔检测方式使其难以融合不同尺度的特征。本文提出了 FSSD(特征融合单次多框检测器),一种增强型 SSD,具有新颖且轻量的特征融合模块,能够在仅略微降低速度的情况下显著提升 SSD 的性能。在特征融合模块中,来自不同层、不同尺度的特征被拼接在一起,随后经过若干下采样模块生成新的特征金字塔,再输入多框检测器以预测最终检测结果。在 Pascal VOC 2007 测试集上,使用单块 Nvidia 1080Ti GPU、输入尺寸为 300×300 时,我们的网络可达到 82.7 mAP(平均精度均值),速度为 65.8 FPS(帧每秒)。此外,我们在 COCO 上的结果也大幅优于传统 SSD。我们的 FSSD 在精度和速度两方面均超越了许多最先进的目标检测算法。代码见 https://github.com/lzx1413/CAFFE_SSD/tree/fssd。

英文摘要:

SSD (Single Shot Multibox Detector) is one of the best object detection algorithms with both high accuracy and fast speed. However, SSD's feature pyramid detection method makes it hard to fuse the features from different scales. In this paper, we proposed FSSD (Feature Fusion Single Shot Multibox Detector), an enhanced SSD with a novel and lightweight feature fusion module which can improve the performance significantly over SSD with just a little speed drop. In the feature fusion module, features from different layers with different scales are concatenated together, followed by some down-sampling blocks to generate new feature pyramid, which will be fed to multibox detectors to predict the final detection results. On the Pascal VOC 2007 test, our network can achieve 82.7 mAP (mean average precision) at the speed of 65.8 FPS (frame per second) with the input size 300$\times$300 using a single Nvidia 1080Ti GPU. In addition, our result on COCO is also better than the conventional SSD with a large margin. Our FSSD outperforms a lot of state-of-the-art object detection algorithms in both aspects of accuracy and speed. Code is available at https://github.com/lzx1413/CAFFE_SSD/tree/fssd.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑