arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2004.06333cs.CV

WQT and DG-YOLO: towards domain generalization in underwater object detection

Peking University(北京大学);Peng Cheng Laboratory(鹏城实验室)

Hong Liu, Pinhao Song, Runwei Ding

英文摘要:

A General Underwater Object Detector (GUOD) should perform well on most of underwater circumstances. However, with limited underwater dataset, conventional object detection methods suffer from domain shift severely. This paper aims to build a GUOD with small underwater dataset with limited types of water quality. First, we propose a data augmentation method Water Quality Transfer (WQT) to increase domain diversity of the original small dataset. Second, for mining the semantic information from data generated by WQT, DG-YOLO is proposed, which consists of three parts: YOLOv3, DIM and IRM penalty. Finally, experiments on original and synthetic URPC2019 dataset prove that WQT+DG-YOLO achieves promising performance of domain generalization in underwater object detection.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑