Laser & Optoelectronics Progress, Volume. 58, Issue 14, 1410019(2021)

Real-Time Target Detection of Underwater Relics Based on Multigranularity Pruning

Youbo Zhang1,2, Wei Guo2,3、*, Yue Zhou1, Gaofei Xu2, Guangwei Li2, and Hongming Sun2,3
Author Affiliations
  • 1College of Engineering Science and Technology, Shanghai Ocean University, Shanghai 201306, China
  • 2Institute of Deep-Sea Science and Engineering, Chinese Academy of Sciences, Sanya, Hainan 572000, China
  • 3University of Chinese Academy of Sciences, Beijing 100049, China
  • show less

    An effective solution for data enhancement, video enhancement, and target detection algorithm compression is proposed to address the data deficiency, blurred images, and limited computing power of an embedded system of the unmanned underwater vehicle for searching underwater relics. First, the data augmentation was used to improve the generalization ability of small sample, and knowledge transfer was then implemented through transfer learning to accelerate the convergence of models. Then, key frame images were selected from the captured video based on the structural similarity, and the selected key frames were enhanced using the contrast limited adaptive histogram equalization stretching in real time. Finally, based on the multigranularity pruning strategy, YOLOV4 performed bidirectional compression of the channel and convolutional layer. The experimental results show that the operation complexity (BFLOPS) of the compressed YOLOV4 model is reduced to 10.588. The average detection speed for images with a size of 640 pixel×480 pixel on the Jetson TX2 embedded image processor reaches 18.2 frame/s.

    Tools

    Get Citation

    Copy Citation Text

    Youbo Zhang, Wei Guo, Yue Zhou, Gaofei Xu, Guangwei Li, Hongming Sun. Real-Time Target Detection of Underwater Relics Based on Multigranularity Pruning[J]. Laser & Optoelectronics Progress, 2021, 58(14): 1410019

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Image Processing

    Received: Oct. 20, 2020

    Accepted: Dec. 2, 2020

    Published Online: Jun. 30, 2021

    The Author Email: Guo Wei (guow@idsse.ac.cn)

    DOI:10.3788/LOP202158.1410019

    Topics