2017 · 6.9K citations · 45 references
ComputingComputer ScienceComputer VisionMachine VisionRobot VisionDeep LearningObject DetectionObject RecognitionScene UnderstandingShape DeformationSpatial TransformationsConvolutionConvolutional Neural NetworksSpatial Feature TransformationTransformations
Convolutional neural networks are limited in modeling geometric transformations because their modules have fixed geometric structures. This work introduces deformable convolution and deformable RoI pooling modules to enhance CNNs’ ability to model transformations. The modules augment spatial sampling with learnable offsets, replace standard counterparts, and can be trained end‑to‑end via back‑propagation. Extensive experiments show that learning dense spatial transformations improves object detection and semantic segmentation, and the code is publicly available.
Convolutional neural networks (CNNs) are inherently limited to model geometric transformations due to the fixed geometric structures in their building modules. In this work, we introduce two new modules to enhance the transformation modeling capability of CNNs, namely, deformable convolution and deformable RoI pooling. Both are based on the idea of augmenting the spatial sampling locations in the modules with additional offsets and learning the offsets from the target tasks, without additional supervision. The new modules can readily replace their plain counterparts in existing CNNs and can be easily trained end-to-end by standard back-propagation, giving rise to deformable convolutional networks. Extensive experiments validate the performance of our approach. For the first time, we show that learning dense spatial transformation in deep CNNs is effective for sophisticated vision tasks such as object detection and semantic segmentation. The code is released at https://github.com/msracver/Deformable-ConvNets.
45
Deep Residual Learning for Image Recognition
Kaiming He, Xiangyu Zhang, Shaoqing Ren et al. · 2016 · 222.1K citations · Full text
ImageNet: A large-scale hierarchical image database
Jia Deng, Wei Dong, Richard Socher et al. · 2009 IEEE Conference on Computer Vision and Pattern Recognition · 2009 · 61.8K citations
Fully convolutional networks for semantic segmentation
Jonathan Long, Evan Shelhamer, Trevor Darrell · 2015 · 36.9K citations