• Laser & Optoelectronics Progress
  • Vol. 58, Issue 14, 1410019 (2021)
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
    DOI: 10.3788/LOP202158.1410019 Cite this Article Set citation alerts
    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 Copy Citation Text show less

    Abstract

    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.
    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