Is Underwater Image Enhancement All Object Detectors Need?


Yudong Wang           Jichang Guo           Wanru He           Huan Gao           Huihui Yue           Zenan Zhang           Chongyi Li
Tianjin University,          S-Lab, Nanyang Technological University

Samples of the visual detection results.

Abstract


Underwater object detection is a crucial and challenging problem in marine engineering and aquatic robot. The difficulty is partly because of the degradation of underwater images caused by light selective absorption and scattering. Intuitively, enhancing underwater images can benefit high-level applications like underwater object detection. However, it is still unclear whether all object detectors need underwater image enhancement as pre-processing. We therefore wonder “Does underwater image enhancement really improve underwater object detection?” and “How does underwater image enhancement contribute to underwater object detection?”. With these two questions, we conduct extensive studies. Specifically, we use 13 state-of-the-art underwater image enhancement algorithms, covering traditional, CNN-based, and GAN-based algorithms, to pre-process underwater object detection data. Then, we retrain 7 popular deep learning-based object detectors using the corresponding results enhanced by different algorithms, obtaining 91 underwater object detection models. Coupled with 7 object detection models retrained using raw underwater images, we employ these 98 models to comprehensively analyze the effect of underwater image enhancement on underwater object detection. We expect this study can provide sufficient exploration to answer the aforementioned questions and draw more attention of the community to the joint problem of underwater image enhancement and underwater object detection. The pre-trained models and results are publicly available and will be regularly updated.

Results

Precision-recall curve at IoU threshold=0.5.

The PR curves (IoU threshold=0.5) of Faster R-CNN with different backbones and training schedules, RetinaNet, Cascade R-CNN, FCOS, ATSS, TOOD, and SSD with average PR curve of 4 categories.

Video Demo



Materials




Paper


Code

Citation

If you find our dataset and paper useful for your research, please consider citing our work:

@inproceedings{Wang2022,
          author = {Wang, Yudong and Guo, Jichang and He, Wanru and Gao, Huan and Yue, Huihui and Zhang, Zenan and Li, Chongyi},
          title = {Is Underwater Image Enhancement All Object Detectors Need?},
          booktitle = {Arixv},
          year = {2022}
          }

License

We retain all the copyrights of this method.

Contact

If you have any question, please contact us via yudongwang@tju.edu.cn.