ABSTRACT
Infrared small target detection is a key technology in infrared search and tracking systems. Despite the rapid development of computer vision in recent years, infrared small target detection based on deep learning is still few and far between due to the lack of texture information and detailed feature of infrared small target. In this paper, we propose an improved U-Net for infrared small target detection, which adds a multi-scale feature fusion module on top of the U-Net. In order to better capture the key information of infrared small target, the multi-scale feature module fuses the high-level semantic information in the convolutional neural network and the low-level small target detail feature across layers. Experimental results show that our proposed method is able to segment infrared small target well and reduce false alarms caused by complex backgrounds compared to traditional methods. In addition, our proposed method performs better than the original U-Net, demonstrating the effectiveness of our proposed improved U-Net.
- Rawat, S., Verma, S.K., and Kumar, Y. 2020. Review on recent development in infrared small target detection algorithms. Procedia Computer Science, 167, 2496-2505.Google ScholarCross Ref
- Zhao, M., Cheng, L., Yang, X., Feng, P., Liu, L., and Wu, N. 2020. TBC-Net: A real-time detector for infrared small target detection using semantic constraint. ArXiv, abs/2001.05852.Google Scholar
- Qi, S., Ma, J., Tao, C., Yang, C., and Tian, J. 2013. A Robust Directional Saliency-Based Method for Infrared Small-Target Detection Under Various Complex Backgrounds. IEEE Geoscience and Remote Sensing Letters, 10, 495-499.Google ScholarCross Ref
- Qian, K., Zhou, H., Rong, S., Wang, B., and Cheng, K. 2017. Infrared dim-small target tracking via singular value decomposition and improved Kernelized correlation filter. Infrared Physics & Technology, 82, 18-27.Google ScholarCross Ref
- Yang, J., Gu, Y., Sun, Z., and Cui, Z. 2019. A Small Infrared Target Detection Method Using Adaptive Local Contrast Measurement. 2019 IEEE International Instrumentation and Measurement Technology Conference (I2MTC), 1-6.Google Scholar
- Chen, C., Li, H., Wei, Y., Xia, T., and Tang, Y. 2014. A Local Contrast Method for Small Infrared Target Detection. IEEE Transactions on Geoscience and Remote Sensing, 52, 574-581.Google ScholarCross Ref
- Deng, Q., Lu, H., Tao, H., Hu, M., and Zhao, F. 2019. Multi-Scale Convolutional Neural Networks for Space Infrared Point Objects Discrimination. IEEE Access, 7, 28113-28123.Google ScholarCross Ref
- Shi, M., and Wang, H. 2020. Infrared Dim and Small Target Detection Based on Denoising Autoencoder Network. Mob. Networks Appl., 25, 1469-1483.Google ScholarDigital Library
- Gao, Z., Dai, J., and Xie, C. 2019. Dim and small target detection based on feature mapping neural networks. J. Vis. Commun. Image Represent., 62, 206-216.Google ScholarDigital Library
- Wang, K., Li, S., Niu, S., and Zhang, K. 2019. Detection of Infrared Small Targets Using Feature Fusion Convolutional Network. IEEE Access, 7, 146081-146092.Google ScholarCross Ref
- Dai, Y., Wu, Y., Zhou, F., and Barnard, K. 2020. Asymmetric Contextual Modulation for Infrared Small Target Detection. ArXiv, abs/2009.14530.Google Scholar
- Dai, Y., Wu, Y., Zhou, F., and Barnard, K. 2020. Attentional Local Contrast Networks for Infrared Small Target Detection. ArXiv, abs/2012.08573.Google Scholar
- Ronneberger, O., Fischer, P., and Brox, T. 2015. U-Net: Convolutional Networks for Biomedical Image Segmentation. ArXiv, abs/1505.04597.Google Scholar
- Cao, Y., Xu, J., Lin, S., Wei, F., and Hu, H. 2019. GCNet: Non-Local Networks Meet Squeeze-Excitation Networks and Beyond. 2019 IEEE/CVF International Conference on Computer Vision Workshop (ICCVW), 1971-1980.Google Scholar
- Rahman, M.A., and Wang, Y. 2016. Optimizing Intersection-Over-Union in Deep Neural Networks for Image Segmentation. International symposium on visual computing. Springer, Cham.Google Scholar
- Wei, Y., You, X., and Li, H. 2016. Multiscale patch-based contrast measure for small infrared target detection. Pattern Recognit., 58, 216-226.Google ScholarDigital Library
- Gao, C., Meng, D., Yang, Y., Wang, Y., Zhou, X., and Hauptmann, A. 2013. Infrared Patch-Image Model for Small Target Detection in a Single Image. IEEE Transactions on Image Processing, 22, 4996-5009.Google ScholarDigital Library
- Dai, Y., and Wu, Y. 2017. Reweighted Infrared Patch-Tensor Model With Both Nonlocal and Local Priors for Single-Frame Small Target Detection. IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing, 10, 3752-3767.Google ScholarCross Ref
- Zhang, L., Peng, L., Zhang, T., Cao, S., and Peng, Z. 2018. Infrared Small Target Detection via Non-Convex Rank Approximation Minimization Joint l2, 1 Norm. Remote. Sens., 10, 1821.Google Scholar
Index Terms
- Improved U-Net Network for Infrared Small Target Detection
Recommendations
Multiple Frames Based Infrared Small Target Detection Method Using CNN
ACAI '21: Proceedings of the 2021 4th International Conference on Algorithms, Computing and Artificial IntelligenceIn the field of infrared, the detection of dim small target has been a challenging topic. Especially in the complex background, a large number of false alarms might appear when using traditional detection methods, this cannot meet the dual requirements ...
Nested Double Pyramid Network for Infrared Small Target Detection
ACAI '21: Proceedings of the 2021 4th International Conference on Algorithms, Computing and Artificial IntelligenceIn this paper a segmentation based single-frame infrared small target detection (SIRST) network is proposed. This nested double pyramid network (NDPNet) is aiming to solve the conflict between receptive field and target size that limited the existing ...
Dim Target Trajectory Recognition Method in Infrared Sequence Images Based on Deep Learning
ACAI '18: Proceedings of the 2018 International Conference on Algorithms, Computing and Artificial IntelligenceAiming at the false targets appearing in the detection of moving weak targets in infrared images, this paper proposes a target trajectory recognition method in infrared sequence images based on deep learning. The method compares the motion law of the ...
Comments