Abstract
Focusing on the problems of CenterNet in infrared images, such as feature loss and insufficient information utilization, an improved algorithm based on spatial feature enhancement is proposed. Firstly, a frequency-space enhancement module is used to enhance the details of the target region. Secondly, a module that can count global information is introduced into the backbone network to model the feature graph globally. Finally, in the case of no increase in computation and complexity, the residual mechanism is adopted to redesign the overall structure of the algorithm, which strengthens the feature interaction simply and efficiently. Experimental tests are carried out on the self-established infrared object detection dataset G-TIR and public infrared object detection dataset FLIR. The proposed algorithm improves the accuracy of the baseline by 8.4% and 15.3% respectively, and is better than many mainstream object detection algorithms in recent years. Meanwhile, the detection speed reaches 72 FPS, which balances the detection accuracy and speed well, then meet the real-time detection requirements.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
Zhou, Y., Tuzel, O.: Voxelnet: end-to-end learning for point cloud based 3d object detection. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 4490–4499 (2018)
Girshick, R., Donahue, J., Darrell, T., Malik, J.: Rich feature hierarchies for accurate object detection and semantic segmentation. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 580–587 (2014)
Ren, S., He, K., Girshick, R., Sun, J.: Faster R-CNN: towards real-time object detection with region proposal networks. IEEE Trans. Pattern Anal. Mach. Intell. 39(6), 1137–1149 (2016)
Bochkovskiy, A., Wang, C., Liao, H.: Yolov4: Optimal speed and accuracy of object detection (2020). arXiv preprint arXiv: 2004.10934
Ge, Z., Liu, S., Wang, F., Li, Z., Sun, J.: Yolox: Exceeding yolo series in 2021 (2021). arXiv preprint arXiv: 2107.08430
Cai, Y., et al.: Yolobile: Real-time object detection on mobile devices via compression-compilation co-design (2020). arXiv preprint arXiv:2009.05697
Zhou, X., Wang, D., Krähenbühl, P.: Objects as points (2019). arXiv preprint arXiv:1904.07850
Chen, Y., Xie, H., Shin, H.: Multi-layer fusion techniques using a CNN for multispectral pedestrian detection. IET Comput. Vision 12(8), 1179–1187 (2018)
Liu, J., Zhang, S., Wang, S., Metaxas, D.N.: Multispectral deep neural networks for pedestrian detection (2016). arXiv preprint arXiv:1611.02644
Ghose, D., Desai, S.M., Bhattacharya, S., Chakraborty, D., Fiterau, M., Rahman, T.: Pedestrian detection in thermal images using saliency maps. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition Workshops, p. 0 (2019)
Wei, S., Wang, C., Chen, Z., Zhang, C., Zhang, X.: Infrared dim target detection based on human visual mechanism. Acta Photonica Sinica 50(1), 0110001 (2021)
Dai, X., Yuan, X., Wei, X.: TIRNet: object detection in thermal infrared images for autonomous driving. Appl. Intell. 51(3), 1244–1261 (2020). https://doi.org/10.1007/s10489-020-01882-2
Zhu, X., Hu, H., Lin, S., Dai, J.: Deformable convnets v2: more deformable, better results. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 9308–9316 (2019)
Yi, J., Wu, P., Liu, B., Huang, Q., Qu, H., Metaxas, D.: Oriented object detection in aerial images with box boundary-aware vectors. In: Proceedings of the IEEE Winter Conference on Applications of Computer Vision, pp. 2150–2159 (2021)
Qin, Z., Zhang, P., Wu, F., Li, X.: Fcanet: frequency channel attention networks. In: Proceedings of the IEEE International Conference on Computer Vision, pp. 783–792 (2021)
Srinivas, A., Lin, T.Y., Parmar, N., Shlens, J., Abbeel, P., Vaswani, A.: Bottleneck transformers for visual recognition. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 16519–16529 (2021)
FLIR Homepage. https://www.flir.cn/oem/adas/adas-dataset-form. Accessed 12 Apr 2022
Zhang, S., Chi, C., Yao, Y., Lei, Z., Li, S.Z.: Bridging the gap between anchor-based and anchor-free detection via adaptive training sample selection. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 9759–9768 (2020)
Tian, Z., Shen, C., Chen, H.: Fcos: fully convolutional one-stage object detection. In: Proceedings of the IEEE International Conference on Computer Vision, pp. 9627–9636 (2019)
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2022 The Author(s), under exclusive license to Springer Nature Switzerland AG
About this paper
Cite this paper
Guo, H., Hou, Z., Sun, Y., Li, J., Ma, S. (2022). Infrared Object Detection Algorithm Based on Spatial Feature Enhancement. In: Yu, S., et al. Pattern Recognition and Computer Vision. PRCV 2022. Lecture Notes in Computer Science, vol 13537. Springer, Cham. https://doi.org/10.1007/978-3-031-18916-6_28
Download citation
DOI: https://doi.org/10.1007/978-3-031-18916-6_28
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-031-18915-9
Online ISBN: 978-3-031-18916-6
eBook Packages: Computer ScienceComputer Science (R0)