[深度学习论文笔记][Object Detection] Fast R-CNN

最新推荐文章于 2024-06-18 14:18:55 发布

Hao_Zhang_Vision

最新推荐文章于 2024-06-18 14:18:55 发布

阅读量698

点赞数

分类专栏： CNN Papers 文章标签： Deep Learning CNN Object Detection Convolutional Neural Papers

本文链接：https://blog.youkuaiyun.com/Hao_Zhang_Vision/article/details/53116068

版权

CNN Papers 专栏收录该内容

58 篇文章

订阅专栏

Fast R-CNN针对R-CNN存在的问题进行改进，提出RoI Pooling层实现固定大小特征映射，使用Smooth l1 Loss减少梯度爆炸，并在训练中采用多任务学习策略提高检测精度。通过采样策略和数据增强手段优化训练过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Girshick, Ross. “Fast r-cnn.” Proceedings of the IEEE International Conference on Computer Vision. 2015. (Citations: 444).

1 R-CNN Problems

• Slow at test-time: need to run full forward pass of CNN for each region proposal.
• Complex multistage training pipeline. Additional SVM and regressor need to be trained.
• SVMs and regressors are post-hoc: CNN features not updated in response to SVMs and regressors.

2 Pipeline

See Fig. The pipeline is as followings:

1. Input. Input an entire image and a set of region proposals (2 k).
2. Forward through conv5. Forward the whole image through CNN to get conv5 feature map.
3. RoI pooling. For each region proposal a region of interest (RoI) pooling layer extracts a fixed-size feature map from the conv5 feature map.
4. Classification and regresson. Each RoI feature map is fed into a sequence of fc layers that finally branch into two sibling output layers: one for classification and another one

for regression. NMS is then performed indepently for each class.

Compared with R-CNN,
• It shares computation of convolutional layers between proposals for an image.
• The whole system is trained end-to-end all at once.

• No disk storage is required for feature caching.

3 RoI Pooling
Suppose conv5 feature map has size D l−1 × H l−1 × W l−1 . We project the region proposal (with sizes F H × F W ) onto conv5 feature map. The fc layers expect fixed size feature map: D l × H l × W l .

The RoI pooling is computed by dividing the projected region proposal into H l × W l grid, and then doing max pooling with each grid cell. RoI pooling can be back propagated
similar to standard max pooling.

4 Smooth l 1 Loss
We use a smooth l 1 loss for bounding box regression.

It is less sensitive to outliers than the l 2 loss used in R-CNN. When the regression targets are unbounded, training with l 2 loss can require careful tuning of learning rates in order to prevent exploding gradients.

5 Training Details
During fine-tuning, we sample 64 RoIs from each image. we take 25% of the RoIs from object proposals that have IoU overlap with a ground-truth bounding box of at least 0.5.
These RoIs comprise the examples labeled with a foreground object class. The remaining RoIs are sampled from object proposals that have a maximum IoU with ground truth in the interval [0.1, 0.5). These are the background examples. The lower threshold of 0.1 appears to act as a heuristic for hard example mining. During training, images are horizontally flipped with probability 0.5. No other data augmentation is used.

6 Results

See Tab. VGG-16 CNN is used on VOC-07 dataset.

Multi-task training improves pure classification accuracy relative to training for classification alone. The improvement ranges from +0.8 to +1.1 mAP points.