人工智能
计算机科学
召回
杂乱
忠诚
召回率
编码(集合论)
人工神经网络
精确性和召回率
探测器
计算机视觉
模式识别(心理学)
雷达
电信
哲学
集合(抽象数据类型)
程序设计语言
语言学
作者
Jingchao Peng,Haitao Zhao,Kaijie Zhao,Zhongze Wang,Lujian Yao
标识
DOI:10.1016/j.eswa.2023.120996
摘要
Infrared small-target detection (ISTD) is an important computer vision task. ISTD aims at separating small targets from complex background clutter. The infrared radiation decays with distance, making the targets highly dim and prone to confusion with the background clutter, which makes the detector challenging to balance the precision and recall rates. To deal with this difficulty, this paper proposes a neural-network-based ISTD method called CourtNet, which has three sub-networks: the prosecution network is designed to improve the recall rate; the defendant network is devoted to increasing the precision rate; the jury network weights their results to adaptively balance the precision and recall rates. CourtNet takes the structure of Transformers, whose feature resolution remains unchanged. Furthermore, the prosecution network utilizes a densely connected structure, which can prevent small targets from disappearing in the forward propagation. In addition, a fine-grained attention module performs attention inside patches to accurately locate the small targets. This paper implements extensive experiments on two ISTD datasets, MFIRST and SIRST, and compares CourtNet with ten other traditional and deep-learning-based methods. Experimental results show that with the fast detection speed (60.61 FPS), CourtNet achieves the best F1 score, 0.62 (in MFIRST) and 0.73 (in SIRST), among the compared methods. The code and dataset will be available at https://github.com/PengJingchao/CourtNet.
科研通智能强力驱动
Strongly Powered by AbleSci AI