基于任务对齐学习的红外船舶目标检测方法

doi:10.12305/j.issn.1001-506X.2025.01.04

Abstract

Abstract:

A task alignment learning (TAL) based infrared ship target detection method is proposed to address the issues of poor detection performance and difficulty in meeting task requirements for multi-scale, small targets, and occlusion in different scenarios during the infrared ship detection process. Firstly, to improve detection speed, an anchor free design is adopted to reduce computational complexity. Then, to improve detection accuracy, TAL is used for label allocation and alignment. Finally, design detection heads tailored to specific scenarios to improve network detection performance. Through experimental comparison and verification, the results show that the proposed method effectively improves the detection performance of ship targets in different scenarios, and outperforms other similar methods in terms of detection accuracy and real-time performance.

Key words: infrared vessel, object detection, anchor-free, task alignment learning (TAL)

CLC Number:

TP391.4

Jie JIANG, Limin ZHANG, Kai LIU, Wenjun YAN, Meng WANG. Infrared ship target detection method based on task alignment learning[J]. Systems Engineering and Electronics, 2025, 47(1): 34-40.

Figures/Tables 11

Fig.1

Fig.2

Fig.3

Fig.4

Fig.5

Table 1

Table 2

Table 3

Fig.6

Table 4

Fig.7

References 30

1	陈钱, 隋修宝. 红外图像处理理论与技术[M]. 北京: 电子工业出版社, 2018.
	CHEN Q , SUI X B . Infrared image processing theory and technology[M]. Beijing: Publishing House of Electronics Industry, 2018.
2	REKAVANDI A M, RASHIDI S, BOUSSAID F, et al. Transformers in small object detection: a benchmark and survey of state-of-the-art[EB/OL]. [2023-09-01], https://arxiv.org/abs/2309.04902.
3	HASSAN S A, RAHIM T, SHIN S Y. Real-time UAV detection based on deep learning network[C]//Proc. of the International Conference on Information and Communication Technology Convergence, 2019.
4	OPROMALLA R , FASANO G , ACCARDO D . A vision-based approach to UAV detection and tracking in cooperative applications[J]. Sensors, 2018, 18 (10):3391. doi: 10.3390/s18103391
5	KOU R , WANG C P , PENG Z M , et al. Infrared small target segmentation networks: a survey[J]. Pattern Recognition, 2023, 143, 109788. doi: 10.1016/j.patcog.2023.109788
6	BAI X Z , CHEN Z G , ZHANG Y , et al. Infrared ship target segmentation based on spatial information improved FCM[J]. IEEE Trans. on Cybernetics, 2015, 46 (12): 3259-3271.
7	LIU Z Y , ZHOU F , CHEN X W , et al. Iterative infrared ship target segmentation based on multiple features[J]. Pattern Recognition, 2014, 47 (9): 2839- 2852. doi: 10.1016/j.patcog.2014.03.005
8	DAI Y M , WU Y Q , ZHOU F , et al. Attentional local contrast networks for infrared small target detection[J]. IEEE Trans. on Geoscience and Remote Sensing, 2021, 59 (11): 9813- 9824. doi: 10.1109/TGRS.2020.3044958
9	甘春生. 星载遥感图像舰船检测方法研究[D]. 辽宁: 沈阳航空航天大学, 2016.
	GAN C S. Research on ship detection methods in spaceborne remote sensing images[D]. Liaoning: Shenyang University of Aeronautics and Astronautics, 2016.
10	LEI S , ZOU Z X , LIU D G , et al. Sea-land segmentation for infrared remote sensing images based on superpixels and multi-scale features[J]. Infrared Physics & Technology, 2018, 91, 12- 17.
11	BAI X Z , ZHOU F G . Analysis of new top-hat transformation and the application for infrared dim small target detection[J]. Pattern Recognition, 2010, 43 (6): 2145- 2156. doi: 10.1016/j.patcog.2009.12.023
12	LIU R M , LU Y H , GONG C L , et al. Infrared point target detection with improved template matching[J]. Infrared Phy-sics & Technology, 2012, 55 (4): 380- 387.
13	李海军, 孔繁程, 林云. 基于改进YOLOv5s的红外舰船检测算法[J]. 系统工程与电子技术, 2023, 45 (8): 2415- 2422.
	LI H J , KONG F C , LIN Y . Infrared ship detection algorithm based on improved YOLOv5s[J]. Systems Engineering and Electronics, 2023, 45 (8): 2415- 2422.
14	潘为年. 基于深度学习的红外成像舰船目标检测方法研究[D]. 成都: 电子科技大学, 2021.
	PAN W N. Research on infrared imaging ship target detection method based on deep learning[D]. Chengdu: University of Electronic Science and Technology of China, 2021.
15	FENG C J, ZHONG Y J, GAO Y, et al. Tood: task-aligned one-stage object detection[C]//Proc. of the IEEE/CVF International Conference on Computer Vision, 2021: 3490-3499.
16	THECKEDATH D , SEDAMKAR R R . Detecting affect states using VGG16, ResNet50 and SE-ResNet50 networks[J]. SN Computer Science, 2020, 1 (2): 79. doi: 10.1007/s42979-020-0114-9
17	TIAN Z , CHEN C H , CHEN H , et al. FCOS: a simple and strong anchor-free object detector[J]. IEEE Trans. on Pattern Analysis and Machine Intelligence, 2020, 44 (4): 1922- 1933.
18	ZHANG Y C , ZHANG W B , YU J Y , et al. Complete and accurate holly fruits counting using YOLOX object detection[J]. Computers and Electronics in Agriculture, 2022, 198, 107062. doi: 10.1016/j.compag.2022.107062
19	WANG C Y, LIAO H Y M, WU Y H, et al. CSPNet: a new backbone that can enhance learning capability of CNN[C]//Proc. of the IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops, 2020: 390-391.
20	JU C , GUAN C T . Tensor-CSPNet: a novel geometric deep learning framework for motor imagery classification[J]. IEEE Trans. on Neural Networks and Learning Systems, 2022, 9, 4512- 4516.
21	XU W J, LONG C J, WANG R S, et al. DRB-GAN: a dyna-mic resblock generative adversarial network for artistic style transfer[C]//Proc. of the IEEE/CVF International Conference on Computer Vision, 2021: 6383-6392.
22	WANG Y J , LEE J G , MOON U C , et al. SSD-TSEFFM: new SSD using trident feature and squeeze and extraction feature fusion[J]. Sensors, 2020, 20 (13): 3630. doi: 10.3390/s20133630
23	DELIANG X , XU Y H , CHENG J D , et al. An agorithm based on a feature interaction-based keypoint detector and sim-CSPNet for SAR image registration[J]. Journal of Radars, 2022, 11 (6): 1081- 1097.
24	ZHANG Y C , ZHANG W B , JI Y Y , et al. Complete and accurate holly fruits counting using YOLOX object detection[J]. Computers and Electronics in Agriculture, 2022, 198, 107062. doi: 10.1016/j.compag.2022.107062
25	DEVER W G . The chronology of Syria-Palestine in the second millennium BCE: a review of current issues[J]. Bulletin of the American Schools of Oriental Research, 1992, 288 (1): 1- 25.
26	SONG G L, LIU Y, WANG X G. Revisiting the sibling head in object detector[C]//Proc. of the IEEE/CVF Conference on Computer Vision and Pattern Recognition. 2020: 11563-11572.
27	FU A M , ZHANG X L , XIONG N X , et al. VFL: a verifiable federated learning with privacy-preserving for big data in industrial IOT[J]. IEEE Trans. on Industrial Informatics, 2020, 18 (5): 3316- 3326.
28	WANG Z H , LIU H , XU S X , et al. A diffraction measurement model and particle filter tracking method for RSS-based DFL[J]. IEEE Journal on Selected Areas in Communications, 2015, 33 (11): 2391- 2403.
29	LI X , WANG W H , WU L J , et al. Generalized focal loss: learning qualified and distributed bounding boxes for dense object detection[J]. Advances in Neural Information Processing Systems, 2020, 33, 21002- 21012.
30	RAZAQ A , HAYAT T , KHAN S A , et al. ATSS model based upon applications of Cattaneo-Christov thermal analysis for entropy optimized ternary nanomaterial flow with homogeneous-heterogeneous chemical reactions[J]. Alexandria Engineering Journal, 2023, 79, 390- 401.

模型	平均准确率均值/%	参数量	运算量	时延/ms	每秒处理帧数
基线模型	48.3	55.58	115.77	15.3	70.3
增加Anchor-free	48.0(-0.3)	53.68	114.78	14.1	70.2
增加CSPResNet	48.7(+0.7)	47.63	101.87	11.9	85.6
增加TAL	49.6(+0.9)	48.42	104.75	12.2	84.1
增加ET-head	50.1(+0.5)	52.31	110.07	13.1	78.3

方法	mAP
ATSS	43.1
SimOTA	44.3
TAL	45.2

算法	mAP1/%	mAP2/%	FPS
YOLOX	80.3	58.9	70.3
YOLOv7	79.9	57.4	65.1
可拓展高效目标检测网络	75.1	52.1	66.7
单发多盒检测网络	81.2	55.6	72.3
区域卷积神经网络	78.1	51.4	62.2
级联卷积神经网络	75.2	52.3	59.8
视网膜网络	74.8	49.8	65.4
全卷积单阶段目标检测	76.6	53.1	76.1
本文网络	82.6	62.3	78.0

训练集与测试集比例	mAP
9∶1	91.7
8∶2	92.3
7∶3	91.5
6∶4	91.2

[1]	Zishuo HAN, Xiquan FAN, Qiang FU, Chuanyan MA, Dongdong ZHANG. Target detection based on multi-source information fusion from the perspective of drones [J]. Systems Engineering and Electronics, 2025, 47(1): 52-61.
[2]	Meng WANG, Bing ZHU. Application of uncertainty modeling in 2D and 3D object detection [J]. Systems Engineering and Electronics, 2023, 45(8): 2370-2376.
[3]	Daozhi WEI, Zhaoyu ZHANG, Jiahao XIE, Ning LI. Multi-sensor cross cueing technique based on improved Actor-Critic algorithm [J]. Systems Engineering and Electronics, 2023, 45(6): 1624-1632.
[4]	Renfei CHEN, Yong PENG, Zhongwen LI. A novel detector for floating objects based on continual unsupervised domain adaptation strategy [J]. Systems Engineering and Electronics, 2023, 45(11): 3391-3401.
[5]	Shuang SONG, Yue ZHANG, Linna ZHANG, Yigang CEN, Yidong LI. Lightweight target detection algorithm based on deep learning [J]. Systems Engineering and Electronics, 2022, 44(9): 2716-2725.
[6]	Hongyao LI, Xiaoqiang LI, Xinzhong HAN, Xueli XIE, Jianxiang XI. Cooperative object detection and recognition algorithm for multiple UAVs based on decision fusion [J]. Systems Engineering and Electronics, 2022, 44(3): 746-754.
[7]	Zizhuang SONG, Jiawei YANG, Dongfang ZHANG, Shiqiang WANG, Shuo ZHANG. Real-time infrared multi-class multi-target anchor-free tracking network [J]. Systems Engineering and Electronics, 2022, 44(2): 401-409.
[8]	Jiali FAN, Shaobing TIAN, Kui HUANG, Xingdong ZHU. Multi-scale object detection algorithm for aircraft carrier surface based on Faster R-CNN [J]. Systems Engineering and Electronics, 2022, 44(1): 40-46.
[9]	Yiming ZHANG, Jianliang AI. Positioning of aerial refueling drogue and docking control based on binocular vision [J]. Systems Engineering and Electronics, 2021, 43(10): 2940-2953.
[10]	Zhenzhen YANG, Jun LE, Yongpeng YANG, Lu FAN. Object detection algorithm of nonconvex motion-assisted low rank and sparse decomposition [J]. Systems Engineering and Electronics, 2020, 42(6): 1218-1225.
[11]	Juan SU, Long YANG, Hua HUANG, Guodong JIN. Improved SSD algorithm for small-sized SAR ship detection [J]. Systems Engineering and Electronics, 2020, 42(5): 1026-1034.
[12]	ZHOU Long, WEI Suyuan, CUI Zhongma, FANG Jiaqi, YANG Xiaoting, YANG Long. Multiobjective detection of complex background radar imagebased on deep learning [J]. Systems Engineering and Electronics, 2019, 41(6): 1258-1264.
[13]	GUO Shao-jun, SHEN Tong-sheng, XU Jian, MA Xin-xing. Detection of multi-ship targets at sea based on ObjectNess BING [J]. Systems Engineering and Electronics, 2016, 38(1): 14-20.
[14]	ZHOU Jun-jing, DUAN Jian-min. Moving object detection for intelligent vehicles based on occupancy grid map [J]. Systems Engineering and Electronics, 2015, 37(2): 436-442.
[15]	DI Xiao-guang, FENG Xiao-dan, Fu Shao-wen. Moving object detection in non-planar scenes based on improved gradient suppression [J]. Systems Engineering and Electronics, 2014, 36(6): 1021-1026.

Infrared ship target detection method based on task alignment learning

RichHTML

PDF (PC)

Knowledge

Abstract

Cite this article

share this article

Figures/Tables 11

References 30

Related Articles 15

Recommended Articles

Metrics

Comments