Abstract
With the wide application of Synthetic Aperture Radar(SAR) radar in maritime surveillance, ship detection method has been developed rapidly. However, there is still a key problem that is the general parallel detection head network has task inconsistency between different tasks. In this work, we proposes a task awareness focal Detector (TAFDet) that alleviate the problem of task inconsistency, consisting of two sub-networks: task awareness attention(TAA-Subnet) and cross-task focal sub-network (CTF-Subnet). Firstly, we design TAA-Subnet that can use the standard backpropagation operation to transform the gradient signal into an attention map of different tasks. To select prominent attention features and obtain a task awareness mask, we proposed a mask generation module(MGM) in TAA-Subnet. Second, CTF-Subnet is designed to tune features in different tasks using task awareness mask, where a feature focal module (FFM) was proposed to enhance the features not paid attention to in the task. Experiments show that our method consistently improved state-of-the-art detectors on SSDD and HRSID datasets. Specifically, our TAFDet consistently surpasses the strong baseline by 0.4\(\sim \)5.8 AP with different backbones on SSDD and HRSID datasets.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
Du, L., Dai, H., Wang, Y., Xie, W., Wang, Z.: Target discrimination based on weakly supervised learning for high-resolution SAR images in complex scenes. IEEE Trans. Geosci. Remote Sens. 58(1), 461–472 (2020)
Shahzad, M., Maurer, M., Fraundorfer, F., Wang, Y., Zhu, X.X.: Buildings detection in VHR SAR images using fully convolution neural networks. IEEE Trans. Geosci. Remote Sens. 57(2), 1100–1116 (2019)
Huang, L., et al.: OpenSARShip: a dataset dedicated to Sentinel-1 ship interpretation. IEEE J. Sel. Topics Appl. Earth Observ. Remote Sens. 11(1), 195–208 (2018)
Zhang, Z., Wang, H., Xu, F., Jin, Y.-Q.: Complex-valued convolutional neural network and its application in polarimetric SAR image classification. IEEE Trans. Geosci. Remote Sens. 55(12), 7177–7188 (2017)
Yang, G., Li, H.-C., Yang, W., Fu, K., Sun, Y.-J., Emery, W.J.: Unsupervised change detection of SAR images based on variational multivariate Gaussian mixture model and Shannon entropy. IEEE Geosci. Remote Sens. Lett. 16(5), 826–830 (2019)
Gierull, C.H.: Demystifying the capability of sublook correlation techniques for vessel detection in SAR imagery. IEEE Trans. Geosci. Remote Sens. 57(4), 2031–2042 (2019)
Ren, S., He, K., Girshick, R., Sun, J.: Faster r-cnn: towards real-time target detection with region proposal networks. IEEE Trans. Pattern Anal. Mach. Intell. 39(6), 1137–1149 (2017)
Redmon, J., Divvala, S., Girshick, R., Farhadi, A.: You only look once: unified, real-time object detection. In: Proceedings of IEEE Conference on Computer Vision and Pattern Recognition, pp. 779–788 (2016)
Liu, W., Anguelov, D., Erhan, D., Szegedy, C., Reed, S., Fu, C.-Y., Berg, A.C.: SSD: single shot multibox detector. In: Leibe, B., Matas, J., Sebe, N., Welling, M. (eds.) ECCV 2016. LNCS, vol. 9905, pp. 21–37. Springer, Cham (2016). https://doi.org/10.1007/978-3-319-46448-0_2
Lin, T., Goyal, P., Girshick, R., He, K., Dollar, P.: Focal loss for dense object detection. IEEE Trans. Pattern Anal. Mach. Intell. 42(2), 318–327 (2020)
Tian, Z., Shen, C., Chen, H., He, T.: FCOS: fully convolutional one-stage object detection. In: Proceedings of IEEE/CVF International Conference on Computer Vision (2019)
Kong, T., et al.: FoveaBox: beyond anchor-based object detector (2019)
Jiang, B., et al.: Acquisition of localization confidence for accurate object detection (2018)
Wang, N., et al.: NAS-FCOS: fast neural architecture search for object detection. In: 2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR). IEEE (2020)
Dai, X., et al.: Dynamic head: Unifying object detection heads with attentions (2021)
Woo, S., Park, J., Lee, J.-Y., Kweon, I.S.: CBAM: convolutional block attention module. In: Ferrari, V., Hebert, M., Sminchisescu, C., Weiss, Y. (eds.) ECCV 2018. LNCS, vol. 11211, pp. 3–19. Springer, Cham (2018). https://doi.org/10.1007/978-3-030-01234-2_1
Huang, Z., Ke, W., Huang, D.: Improving object detection with inverted attention. arXiv:1903.12255 (2019)
Feng, C., et al.: TOOD: task-aligned one-stage object detection (2021)
Li, X., et al.: Generalized focal loss: learning qualified and distributed bounding boxes for dense objectdetection. arXiv preprint arXiv:2006.04388 (2020)
Cao, Y., Chen, K., Loy, C.C., Lin, D.: Prime sample attention in object detection. arXivpreprint arXiv:1904.04821 (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
Lv, Y., Li, M., He, Y., Song, Y. (2022). TAFDet: A Task Awareness Focal Detector for Ship Detection in SAR Images. 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_25
Download citation
DOI: https://doi.org/10.1007/978-3-031-18916-6_25
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)