Command Palette
Search for a command to run...
Hiérarchies riches de descripteurs pour la détection d'objets précise et la segmentation sémantique
Hiérarchies riches de descripteurs pour la détection d'objets précise et la segmentation sémantique
Ross Girshick Jeff Donahue Trevor Darrell Jitendra Malik
Résumé
Les performances en détection d'objets, mesurées sur l'ensemble de données canonique PASCAL VOC, ont stagné ces dernières années. Les méthodes les plus performantes sont des systèmes d'ensemble complexes qui combinent généralement plusieurs descripteurs visuels de bas niveau avec un contexte de haut niveau. Dans cet article, nous proposons un algorithme de détection simple et évolutif qui améliore la précision moyenne (mAP) de plus de 30 % par rapport au meilleur résultat précédent sur VOC 2012, atteignant une mAP de 53,3 %. Notre approche combine deux idées clés : (1) il est possible d'appliquer des réseaux de neurones convolutifs (CNN) à grande capacité à des propositions de régions ascendantes afin de localiser et segmenter les objets et (2) lorsque les données d'entraînement annotées sont rares, un pré-entraînement supervisé sur une tâche auxiliaire, suivi d'un ajustement fin spécifique au domaine, produit une amélioration significative des performances. Puisque nous combinons des propositions de régions avec des CNN, nous appelons notre méthode R-CNN : Regions with CNN features. Nous présentons également des expériences qui éclairent ce que le réseau apprend, révélant une hiérarchie riche de descripteurs visuels. Le code source du système complet est disponible à l'adresse http://www.cs.berkeley.edu/˜rbg/rcnn.
One-sentence Summary
UC Berkeley and ICSI propose R-CNN, a simple and scalable detection algorithm that applies high-capacity convolutional neural networks to bottom-up region proposals for object localization and segmentation, leveraging supervised pre-training with domain-specific fine-tuning when labeled data is scarce to achieve a mAP of 53.3% on PASCAL VOC 2012, a 30% relative improvement over the prior best result.
Key Contributions
-
The paper introduces R-CNN, a simple scalable object detector that applies high-capacity convolutional neural networks to bottom-up region proposals for localization and segmentation, improving mean average precision to 53.3% on PASCAL VOC 2012, a more than 30% relative gain over the previous best result.
-
The work shows that supervised pre-training on a large image classification dataset, followed by domain-specific fine-tuning on PASCAL detection data, is effective when labeled data is scarce; this strategy improves mean average precision by 8 percentage points and reaches 54% on PASCAL VOC 2010 versus 33% for a tuned HOG-based deformable part model.
-
The work provides analysis and extension: probing reveals a rich hierarchy of diverse CNN features, a simple bounding box regression reduces dominant mislocalization errors, and minor modifications achieve state-of-the-art PASCAL VOC segmentation with 47.9% average accuracy.
Introduction
Visual object detection had long relied on handcrafted features such as SIFT and HOG, but performance on benchmarks like PASCAL VOC had plateaued because these low-level descriptors struggle to capture the hierarchical representations needed for robust recognition. Earlier CNNs had shown promise in image classification, yet applying them to detection was difficult: deep networks have large receptive fields that make sliding-window localization awkward, and labeled detection data are too limited to train large CNNs from scratch. The authors introduce R-CNN, which combines category-independent region proposals with CNN features and category-specific linear classifiers to localize and classify objects. They also show that supervised pre-training on large image classification data, followed by fine-tuning on a smaller detection dataset, makes high-capacity CNNs practical for detection and yields major performance gains on PASCAL VOC.
Method
The method follows a modular region-based detection architecture. Instead of training a single end-to-end detector, the authors separate the system into three main components: category-independent region proposal, fixed-length CNN feature extraction, and class-specific linear classification. This decomposition makes the pipeline efficient because the expensive feature computation can be shared across many object classes.
The first module generates candidate object locations. In the described detection system, the authors use selective search to produce about 2000 category-independent region proposals per test image. These proposals define the candidate windows that the detector later evaluates. The architecture is not tied to a specific proposal generator, but selective search provides a controlled and effective set of windows for detection.
The second module converts each proposal into a CNN feature vector. The network requires inputs of size 227×227. To handle proposals of arbitrary size and aspect ratio, the authors extract the tight bounding box around each proposal, dilate it to add a small amount of image context, and warp the result to the required input size. Before warping, the box is expanded so that exactly p=16 pixels of warped context surround the original proposal after resizing. Refer to the figure below.
The warped region is mean-subtracted and passed through a convolutional network consisting of five convolutional layers and two fully connected layers. The output is a 4096-dimensional feature vector for the proposal. This fixed-length representation is then used for scoring, training, and downstream refinement.
At test time, the detection pipeline runs the proposal module once on the image and forward propagates all warped proposals through the shared CNN. The resulting feature matrix is typically of size 2000×4096. For each class, this matrix is multiplied by the corresponding SVM weight matrix, and each proposal receives a score. A greedy non-maximum suppression step then removes lower-scoring overlapping proposals for each class independently. Because the CNN parameters and features are shared across categories, the proposal and feature extraction cost is amortized over all classes. This shared computation is central to the scalability of the method.
The training process is organized into several stages. The authors first perform supervised pre-training of the CNN on a large auxiliary image classification dataset. This provides a generic visual representation that can be transferred to detection. The CNN is then fine-tuned specifically for the detection task and the warped proposal domain. During fine-tuning, the original 1000-way classification layer is replaced with a randomly initialized 21-way layer, corresponding to twenty object classes plus background. The authors treat proposals with IoU overlap of at least 0.5 against a ground-truth box as positive examples for that class, while all other proposals are treated as negatives.
Fine-tuning uses stochastic gradient descent with a relatively small learning rate of 0.001, which is one tenth of the pre-training learning rate. This allows the network to adapt to detection windows without severely disrupting the pre-trained initialization. Each mini-batch has size 128 and is constructed by sampling 32 positive windows and 96 background windows. The sampling is biased toward positive windows because true positives are rare among the full proposal set.
After the CNN is fine-tuned, the authors train class-specific binary SVMs on the extracted CNN features. These SVMs are not simply replaced by the final softmax layer of the fine-tuned network. Positive examples are the ground-truth boxes for each class. Negative examples are proposals whose IoU overlap with the ground truth is below a threshold. The threshold is selected on validation data and is set to 0.3. This choice is important because too loose or too strict a negative definition degrades detection accuracy. Since the full proposal set is large and difficult to fit directly in memory, the authors use hard negative mining to efficiently find the most informative negative examples.
The method also includes an interpretation of the learned CNN representation. To understand what the network has learned, the authors inspect units from the layer pool5, the max-pooled output of the fifth convolutional layer. Each pool unit is treated as a detector over a large set of held-out region proposals. The authors compute the unit’s activations, rank proposals by activation strength, apply non-maximum suppression, and visualize the top-scoring inputs. As shown in the figure below:
This procedure shows that different pool5 units can specialize in different visual patterns. Some units respond to concrete object concepts, while others respond to texture, color, or material-like structure. The result supports the design choice of using deep CNN features: the network learns a representation that combines shape, texture, color, and class-sensitive cues.
Error analysis motivates an additional refinement module. The authors find that many false positives are not caused by confusing one object class with another, but by poor localization of the correct object. To address this, they add a bounding-box regression step. Given the pool5 feature vector for a proposal, a linear regression model predicts an adjusted detection window. Refer to the figure below.
This regression stage tightens loose proposals by transforming them into more accurate windows. It is a lightweight post-processing module that directly targets localization errors, improving the final detection quality without requiring a fully new detection backbone.
The same feature extraction pipeline can also be adapted to semantic segmentation. In that setting, region proposals are warped to the standard network input, and CNN features are computed for each region. The authors compare strategies that use the full warped window, the foreground mask only, or a concatenation of both. The full-window strategy captures surrounding context, while the foreground-mask strategy emphasizes the region shape itself. Combining the two gives the network access to both object appearance and contextual information, enabling the region classification framework to be reused for dense pixel labeling.
Experiment
The experiments evaluate R-CNN on PASCAL VOC object detection and segmentation, validating its design choices through comparisons with strong baselines, layer-wise ablations, feature visualizations, error analysis, and bounding box regression. Detection results show that CNN features greatly outperform prior hand-crafted and learned feature pipelines, while ablations indicate that convolutional features provide much of the representational strength and fine-tuning substantially improves performance. Visualizations reveal interpretable class- and texture-related units, error analysis motivates bounding box regression for better localization, and segmentation experiments demonstrate that CNN region features are competitive with a leading framework, especially when combining full-window and foreground-mask representations.
The results show that the bounding-box regression variant of R-CNN obtains the highest average detection precision among the compared methods, while R-CNN alone also clearly surpasses strong baselines that use selective search proposals. The improvement is broad across object categories and is attributed to reduced localization errors from learned bounding box adjustment. Bounding-box regression raises average detection precision by a small but consistent amount over R-CNN. R-CNN with bounding-box regression outperforms UVA and Regionlets, which use selective search proposals. The method also exceeds SegDPM and DPM, despite those baselines using context rescoring.
On the VOC 2007 test set, detection average precision improves when CNN features are fine-tuned on the VOC trainval data. Without fine-tuning, fc6 features give better average precision than fc7 features and pool5 features, while fc7 features are the best among the fine-tuned feature layers. A bounding box regression stage further increases detection average precision, indicating that localization errors limit earlier R-CNN results. Without fine-tuning, fc6 features are more effective for detection than fc7 features. Fine-tuning improves all three feature layers, and bounding box regression gives an additional accuracy gain.
On the VOC 2011 validation segmentation benchmark, R-CNN features combining full-region and foreground-masked representations produce the strongest mean accuracy, slightly surpassing the O2P baseline. Across all feature strategies, fc6 features consistently outperform fc7 features, while foreground-masked regions are a bit better than full-window regions, indicating that region shape provides useful signal. The combined full and foreground feature representation is the top-performing R-CNN variant for segmentation mean accuracy. Using fc6 features is more effective than using fc7 features across full, foreground, and combined region strategies.
On the VOC 2011 test set, the combined full and foreground CNN feature method achieves the highest mean segmentation accuracy among the compared approaches, surpassing the Regions and Parts baseline and slightly exceeding second-order pooling. It also performs best for many individual categories without any fine-tuning, while foreground-masked features outperform full-window features, and their combination is strongest. The full+fg fc6 method gives the best overall mean segmentation accuracy. It is the top method for 11 of 21 categories without fine-tuning. Foreground-masked features are stronger than full-window features, and combining them improves segmentation.
Experiments on VOC detection and segmentation evaluate R-CNN features with different feature layers, fine-tuning, and bounding-box regression. Detection results show that fine-tuning CNN features improves performance, while learned bounding-box adjustment consistently reduces localization errors and yields the strongest precision among compared methods. Segmentation results show that foreground-masked features outperform full-window features, and combining both strategies with fc6 features achieves the best overall accuracy.