Real-time ship detection in satellite images based on YOLO-v3 model compression

Ke-jun CHEN; Ye ZHANG

doi:10.37188/YJYXS20203511.1168

您当前的位置：

首页 >

文章列表页 >

Real-time ship detection in satellite images based on YOLO-v3 model compression

Image Processing | 更新时间：2020-10-30

- Real-time ship detection in satellite images based on YOLO-v3 model compression
- Chinese Journal of Liquid Crystals and Displays Vol. 35, Issue 11, Pages: 1168-1176(2020)
- 作者机构：
  
  1.中国科学院长春光学精密机械与物理研究所应用光学国家重点实验室, 吉林长春 130033
  2.中国科学院大学, 北京 100049
- 作者简介：
- 基金信息：
  
  Special Zone of National Defense Science and Technology Innovation of the CMC(18-H863-00-TS-002-018-01);Youth Innovation Promotion Association, CAS(2016201)
- DOI：10.37188/YJYXS20203511.1168
  CLC： TP391.4
- Received：25 March 2020，
  
  Revised：23 June 2020，
  
  Accepted：23 June 2020，
  
  Published：05 November 2020
- 稿件说明：
移动端阅览
Ke-jun CHEN, Ye ZHANG. Real-time ship detection in satellite images based on YOLO-v3 model compression[J]. Chinese journal of liquid crystals and displays, 2020, 35(11): 1168-1176.
DOI：

Ke-jun CHEN, Ye ZHANG. Real-time ship detection in satellite images based on YOLO-v3 model compression[J]. Chinese journal of liquid crystals and displays, 2020, 35(11): 1168-1176. DOI： 10.37188/YJYXS20203511.1168.

摘要

常见的目标检测模型由于模型参数量较大，往往难以部署在无人机、卫星等移动嵌入式设备上。为了对船只进行实时监测，将目标检测模型部署在计算能力较弱的设备上，对基于计算机视觉的卫星图像船只目标检测方法进行研究。针对卫星图像中船舰的形状长宽比例特点，采用K-means++聚类算法选取初始的锚点框；接着对模型进行多尺度训练，将多尺度金字塔图像作为模型训练的输入；将YOLO-v3目标检测算法的批归一化层的尺度因子作为通道重要性的度量指标，对YOLO-v3模型进行剪枝压缩。实验结果表明，采用的模型剪枝和压缩方法能有效地对模型进行压缩，模型的参数量减少了91.5%，模型检测时间缩短了60%，极大地减少了系统计算性能的开销。当采用的初始锚点框个数为6个时，平均准确率（mAP）达到77.31%，满足了卫星图像船舰实时性检测的需求。

Abstract

Due to the large number of model parameters

common target detection models were often difficult to be deployed on mobile embedded platforms such as unmanned aerial vehicle and satellite. In order to detect ships in real time

and for the purpose of deploying target detection model in weak computing equipment

the ship detection algorithm based on computer vision was researched. According to the feature of ship shape length ratio and width ratio in satellite images

K-means ++ clustering algorithm was used to select the initial candidate anchor boxes. Multi-scale pyramid images were used as the input of model training. The scale factor of the batch normalization layer of the YOLO-v3 target detection algorithm was taken as the measure index of channel importance

and the YOLO-v3 model was pruned and compressed. Experimental results show that model pruning and compression method can effectively compress the model. The number of parameters of the model size is reduced by 91.5% and the time of model detection is shortened by 60% compared with the original model

which greatly reduces the overhead of system computing performance. When the initial number of candidate boxes is 6

the mAP reaches at 77.31%

which meets the requirements of real-time detecting ship in satellite images.

关键词

Keywords

references

马啸 , 邵利民 , 金鑫 , 等 . 舰船目标识别技术研究进展 . 科技导报 , 2019 . 37 ( 24 ): 65 - 78 . http://www.wanfangdata.com.cn/details/detail.do?_type=perio&id=kjdb201924011 http://www.wanfangdata.com.cn/details/detail.do?_type=perio&id=kjdb201924011 .

X MA , L M SHAO , X JIN , 等 . Advances in ship target recognition technology . Science & Technology Review , 2019 . 37 ( 24 ): 65 - 78 . http://www.wanfangdata.com.cn/details/detail.do?_type=perio&id=kjdb201924011 http://www.wanfangdata.com.cn/details/detail.do?_type=perio&id=kjdb201924011 .

安洁玉 , 丁斌芬 . 无人机海监测绘技术应用下舰船遥感图像目标检测 . 舰船科学技术 , 2019 . 41 ( 24 ): 187 - 189 . http://www.cnki.com.cn/Article/CJFDTotal-JCKX201924064.htm http://www.cnki.com.cn/Article/CJFDTotal-JCKX201924064.htm .

J Y AN , B F DING . The target detection of the ship's remote sensing image under the UAVs . Ship Science and Technology , 2019 . 41 ( 24 ): 187 - 189 . http://www.cnki.com.cn/Article/CJFDTotal-JCKX201924064.htm http://www.cnki.com.cn/Article/CJFDTotal-JCKX201924064.htm .

胡炎 , 单子力 , 高峰 . 基于Faster-RCNN和多分辨率SAR的海上舰船目标检测 . 无线电工程 , 2018 . 48 ( 2 ): 96 - 100 . http://www.cnki.com.cn/Article/CJFDTotal-WXDG201802005.htm http://www.cnki.com.cn/Article/CJFDTotal-WXDG201802005.htm .

Y HU , Z L SHAN , F GAO . Ship detection based on Faster-RCNN and multiresolution SAR . Radio Engineering , 2018 . 48 ( 2 ): 96 - 100 . http://www.cnki.com.cn/Article/CJFDTotal-WXDG201802005.htm http://www.cnki.com.cn/Article/CJFDTotal-WXDG201802005.htm .

REN S Q, HE K M, GIRSHICK R B, et al . Faster R-CNN: towards real-time object detection with region proposal networks[C]// Proceedings of Advances in Neural Information Processing Systems 28. Montreal, Quebec, Canada: NIPS, 2015: 91-99.

REDMON J, DIVVALA S, GIRSHICK R, et al . You only look once: unified, real-time object detection[C]// Proceedings of 2016 IEEE Conference on Computer Vision and Pattern Recognition . Las Vegas, USA: IEEE, 2016: 779-788.

马永杰 , 宋晓凤 . 基于YOLO和嵌入式系统的车流量检测 . 液晶与显示 , 2019 . 34 ( 6 ): 613 - 618 . http://cjlcd.lightpublishing.cn/thesisDetails?columnId=1383321&Fpath=&index=-1&l=zh http://cjlcd.lightpublishing.cn/thesisDetails?columnId=1383321&Fpath=&index=-1&l=zh .

Y J MA , X F SONG . Vehicle flow detection based on YOLO and embedded system . Chinese Journal of Liquid Crystals and Displays , 2019 . 34 ( 6 ): 613 - 618 . http://cjlcd.lightpublishing.cn/thesisDetails?columnId=1383321&Fpath=&index=-1&l=zh http://cjlcd.lightpublishing.cn/thesisDetails?columnId=1383321&Fpath=&index=-1&l=zh .

马啸 , 邵利民 , 金鑫 , 等 . 改进的YOLO模型及其在舰船目标识别中的应用 . 电讯技术 , 2019 . 59 ( 8 ): 869 - 874 . http://www.wanfangdata.com.cn/details/detail.do?_type=perio&id=dianxjs201908001 http://www.wanfangdata.com.cn/details/detail.do?_type=perio&id=dianxjs201908001 .

X MA , L M SHAO , X JIN , 等 . Improved YOLO model and its application in ship target recognition . Telecommunication Engineering , 2019 . 59 ( 8 ): 869 - 874 . http://www.wanfangdata.com.cn/details/detail.do?_type=perio&id=dianxjs201908001 http://www.wanfangdata.com.cn/details/detail.do?_type=perio&id=dianxjs201908001 .

REDMON J, FARHADI A. Yolov3: an incremental improvement[J]. arXiv: 1804.02767, 2018.

REDMON J, FARHADI A. YOLO9000: better, faster, stronger[C]/ /Proceedings of 2017 IEEE Computer Vision and Pattern Recognition . Honolulu, USA: IEEE, 2017: 6517-6525.

M EVERINGHAM , L VAN GOOL , C K I WILLIAMS , 等 . The PASCAL visual object classes (VOC) challenge . International Journal of Computer Vision , 2010 . 88 ( 2 ): 303 - 338 . http://www.wanfangdata.com.cn/details/detail.do?_type=perio&id=455e1aa382285717f41122eeddab162e http://www.wanfangdata.com.cn/details/detail.do?_type=perio&id=455e1aa382285717f41122eeddab162e .

刘梦伦, 赵希梅, 魏宾.基于多尺度多特征卷积神经网络的肝硬化识别[J/OL].计算机仿真[2020-03-08] . http://kns.cnki.net/kcms/detail/11.3724.TP.20191114.1056.062.html http://kns.cnki.net/kcms/detail/11.3724.TP.20191114.1056.062.html .

LIU M L, ZHAO X M, WEI B. Cirrhosis recognition based on multi-scale multi-feature convolutional neural network[J/OL]. Computer Simulation [2020-03-08]. http://kns.cnki.net/kcms/detail/11.3724.TP.20191114.1056.062.html http://kns.cnki.net/kcms/detail/11.3724.TP.20191114.1056.062.html . (in Chinese)

HE K M, ZHANG X Y, REN S Q, et al . Deep residual learning for image recognition[C]// Proceedings of the 2016 IEEE Conference on Computer Vision and Pattern Recognition . Las Vegas, USA: IEEE, 2016: 770-778.

ZHANG P Y, ZHONG Y X, LI X Q. SlimYOLOv3: narrower, faster and better for real-time UAV applications[C]// Proceedings of 2019 IEEE/CVF International Conference on Computer Vision Workshop . Seoul, Korea (South): IEEE, 2019: 37-45.

IOFFE S, SZEGEDY C. Batch normalization: accelerating deep network training by reducing internal covariate shift[C]// Proceedings of the 32 nd International Conference on International Conference on Machine Learning . Lille, France: JMLR, 2015: 448-456.

LIU Z, LI J G, SHEN Z Q, et al . Learning efficient convolutional networks through network slimming[C]// Proceedings of 2017 IEEE International Conference on Computer Vision . Venice, Italy: IEEE, 2017: 2755-2763.

Airbus ship detection challenge[DB/OL]. https://www.kaggle.com/c/airbus-ship-detection https://www.kaggle.com/c/airbus-ship-detection .

SIMONYAN K, ZISSERMAN A. Very deep convolutional networks for large-scale image recognition[C]// Proceedings of the 3 rd International Conference on Learning Representations . San Diego, USA: ICLR, 2015: 1-14.

LIU W, ANGUELOV D, ERHAN D, et al . SSD: single shot multibox detector[C]/ /Proceedings of the 14 th European Conference on Computer Vision . Amsterdam, The Netherlands: Springer, 2016: 21-37.

DAI J, LI Y, HE K, et al . R-FCN: object detection via region-based fully convolutional networks[C]// Proceedings of the 30 th Conference on Neural Information Processing Systems . Barcelona, Spain: NIPS, 2016: 379-387.

Views

下载量

CSCD

Alert me when the article has been cited

提交

Tools

Publicity Resources

Design of aerial image target detection systembased on MPSOC

Related Author

Bin REN

Yu-qing WANG

Zhen CONG

Hai-tao NIE

Hang YANG

Bin REN

Yu-qing WANG

Hai-tao NIE

Related Institution

Changchun Institute of Optics, Final Mechanics and Physics, Chinese Academy of Sciences

Department Medical Engineing, The 964th Hospital, Joint Service Support Unit, Chinese People's Liberation Army

Changchun Institute of Optics ，Final Mechanics and Physics ，Chinese Academy of Sciences

AI问答

Address：No.3888 Dong Nanhu Road, Changchun, Jilin, China 130033 Postal code：130033
Tel：0431-86176059 Email：yjxs@ciomp.ac.cn
Technical support is provided by Beijing Founder electronics co., LTD 吉ICP备11002662号-17 京公网安备11010802024621
It is recommended to read the content of this site in Chrome&IE9+. Please switch to extreme mode in browser 360.
Cookies We use cookies to help provide and enhance our service and tailor content. By continuing, you agree to the use of cookies.

⁰