DeepAI AI Chat
Log In Sign Up

Boosting R-CNN: Reweighting R-CNN Samples by RPN's Error for Underwater Object Detection

by   Pinhao Song, et al.

Complicated underwater environments bring new challenges to object detection, such as unbalanced light conditions, low contrast, occlusion, and mimicry of aquatic organisms. Under these circumstances, the objects captured by the underwater camera will become vague, and the generic detectors often fail on these vague objects. This work aims to solve the problem from two perspectives: uncertainty modeling and hard example mining. We propose a two-stage underwater detector named boosting R-CNN, which comprises three key components. First, a new region proposal network named RetinaRPN is proposed, which provides high-quality proposals and considers objectness and IoU prediction for uncertainty to model the object prior probability. Second, the probabilistic inference pipeline is introduced to combine the first-stage prior uncertainty and the second-stage classification score to model the final detection score. Finally, we propose a new hard example mining method named boosting reweighting. Specifically, when the region proposal network miscalculates the object prior probability for a sample, boosting reweighting will increase the classification loss of the sample in the R-CNN head during training, while reducing the loss of easy samples with accurately estimated priors. Thus, a robust detection head in the second stage can be obtained. During the inference stage, the R-CNN has the capability to rectify the error of the first stage to improve the performance. Comprehensive experiments on two underwater datasets and two generic object detection datasets demonstrate the effectiveness and robustness of our method.


page 1

page 5

page 11

page 14


RoIMix: Proposal-Fusion among Multiple Images for Underwater Object Detection

Generic object detection algorithms have proven their excellent performa...

SWIPENET: Object detection in noisy underwater images

In recent years, deep learning based object detection methods have achie...

Underwater Object Classification and Detection: first results and open challenges

This work reviews the problem of object detection in underwater environm...

C-RPNs: Promoting Object Detection in real world via a Cascade Structure of Region Proposal Networks

Recently, significant progresses have been made in object detection on c...

ProbaNet: Proposal-balanced Network for Object Detection

Candidate object proposals generated by object detectors based on convol...

S-OHEM: Stratified Online Hard Example Mining for Object Detection

One of the major challenges in object detection is to propose detectors ...

Edge-guided Representation Learning for Underwater Object Detection

Underwater object detection (UOD) is crucial for marine economic develop...