Underwater real-time target detection based on key frame and model compression
文献类型:期刊论文
作者 | Guo,Wei1![]() ![]() |
刊名 | Journal of Physics: Conference Series
![]() |
出版日期 | 2021-02-01 |
卷号 | 1800期号:1 |
ISSN号 | 1742-6588 |
DOI | 10.1088/1742-6596/1800/1/012001 |
英文摘要 | Abstract In order to improve the speed and keep high accuracy of UUVs object detection on embedded device. This paper proposes an acceleration strategy based on key frame extraction and model compression. Firstly, selecting shipwreck as target and creating a real-world underwater image dataset of the shipwreck. And this work applies both online and offline data augmentation to accelerate the model convergence and to improve the model generalization. Then, this paper builds a fusion algorithm with structural similarity, color histogram and image entropy to select the key frame of the captured video for removing the redundant information, and analyzing the reason for the complexity of the neural network and the method of model compression. Finally, this work cuts off unimportant structure of network based on channel and layer pruning to reduce model complexity and keeps the model with high accuracy by fine-tuning. The experiment shows the Billion Floating Operations of the compressed YOLOV4 model is reduced to 15.778. On the Nvidia Jetson TX2 embedded image processor, the average detection speed for video with 608 × 608 resolution can reach 15.12FPS. And the detection speed of the video composed of key frames is 2.98 times faster than the raw video. |
语种 | 英语 |
WOS记录号 | IOP:1742-6588-1800-1-012001 |
出版者 | IOP Publishing |
版本 | 出版稿 |
源URL | [http://ir.idsse.ac.cn/handle/183446/8328] ![]() |
专题 | 深海工程技术部 |
作者单位 | 1.Institute of Deep-Sea Science and Engineering, Chinese Academy of Sciences, Hainan, 572000, China 2.College of Engineering Science and Technology, Shanghai Ocean University, Shanghai, 201306, China |
推荐引用方式 GB/T 7714 | Guo,Wei,Zhang,Youbo,Zhou,Yue,et al. Underwater real-time target detection based on key frame and model compression[J]. Journal of Physics: Conference Series,2021,1800(1). |
APA | Guo,Wei,Zhang,Youbo,Zhou,Yue,Xu,Gaofei,&Li,Guangwei.(2021).Underwater real-time target detection based on key frame and model compression.Journal of Physics: Conference Series,1800(1). |
MLA | Guo,Wei,et al."Underwater real-time target detection based on key frame and model compression".Journal of Physics: Conference Series 1800.1(2021). |
入库方式: OAI收割
来源:深海科学与工程研究所
其他版本
除非特别说明,本系统中所有内容都受版权保护,并保留所有权利。