Ship Detection with Lightweight Network Based on YOLOV3

被引:0
作者
Kong, Decheng [1 ]
Wang, Ping [1 ]
Wei, Xiang [1 ]
Xu, Zeyu [1 ]
机构
[1] Harbin Engn Univ, Coll Mech & Elect Engn, Harbin, Peoples R China
来源
INTERNATIONAL CONFERENCE ON MECHANICAL DESIGN AND SIMULATION (MDS 2022) | 2022年 / 12261卷
关键词
ship detection; YOLOV3; lightweight network; unmanned surface vehicle;
D O I
10.1117/12.2640800
中图分类号
TH [机械、仪表工业];
学科分类号
0802 ;
摘要
This paper proposes a lightweight network for ship detection based on YOLOV3. This method reduces the parameters and model size of YOLOV3 while retaining accuracy from three aspects. First, an efficient way is described of replacing the Darknet-53 of YOLOV3 with a reduced network. Second, an optimization form is demonstrated to improve the basic unit. Last, compression and acceleration strategies are implemented in the method. The performance of our method and YOLOV3 is measured on a self-built dataset for ship detection. The method we proposed is 29x less parameters, 85x less size, and 3 x less inference time than YOLOV3. In the experiment, the method is deployed on a Qihang Unmanned Surface Vehicle (USV) and realizes real-time detection of ships.
引用
收藏
页数:6
相关论文
共 15 条
[1]  
[Anonymous], 2018, DISCRIMINATION AWARE
[2]  
Dettmers T, 2016, Arxiv, DOI arXiv:1511.04561
[3]  
Gupta S, 2015, PR MACH LEARN RES, V37, P1737
[4]  
Han S, 2015, ADV NEUR IN, V28
[5]   Deep Residual Learning for Image Recognition [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :770-778
[6]  
Hinton G., 2014, Distilling the knowledge in a neural network
[7]  
Howard Andrew G., 2017, arXiv
[8]   Quantization and Training of Neural Networks for Efficient Integer-Arithmetic-Only Inference [J].
Jacob, Benoit ;
Kligys, Skirmantas ;
Chen, Bo ;
Zhu, Menglong ;
Tang, Matthew ;
Howard, Andrew ;
Adam, Hartwig ;
Kalenichenko, Dmitry .
2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, :2704-2713
[9]   Improving the Interpretability of Deep Neural Networks with Knowledge Distillation [J].
Liu, Xuan ;
Wang, Xiaoguang ;
Matwin, Stan .
2018 18TH IEEE INTERNATIONAL CONFERENCE ON DATA MINING WORKSHOPS (ICDMW), 2018, :905-912
[10]   Learning Efficient Convolutional Networks through Network Slimming [J].
Liu, Zhuang ;
Li, Jianguo ;
Shen, Zhiqiang ;
Huang, Gao ;
Yan, Shoumeng ;
Zhang, Changshui .
2017 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2017, :2755-2763