Fine-grained ship image classification and detection based on a vision transformer and multi-grain feature vector FPN model

Autor: Fengxiang Wang, Deying Yu, Liang Huang, Yalun Zhang, Yongbing Chen, Zhiguo Wang
Jazyk: angličtina
Rok vydání: 2024
Předmět:
Zdroj: Geo-spatial Information Science, Pp 1-22 (2024)
Druh dokumentu: article
ISSN: 10095020
1993-5153
1009-5020
86203231
DOI: 10.1080/10095020.2024.2331552
Popis: ABSTRACTIn naval and civilian domains, meticulous ship classification and detection are paramount. Nevertheless, predominant research has gravitated toward leveraging Convolutional Neural Network (CNN)-centered methodologies, often overlooking the diverse granularity inherent in ship samples. In our pursuit to holistically extract features from ship images across varying granularities, we present a transformative architecture: the Vision Transformer and Multi-Grain Feature Vector Feature Pyramid Network (ViT-MGFV-FPN). This model synergistically melds the merits of MGFV-FPN with an augmented Vision Transformer (ViT) for a comprehensive image feature extraction. To cater to the extraction of broader image features whilst sidestepping the innate quadratic complexity of traditional ViT, we unveil an enhanced version christened the Global Swin Transformer. Concurrently, the MGFV-FPN is orchestrated to harness the prowess of CNNs in distilling intricate ship attributes. Rigorous empirical evaluations underscore our model’s superiority in juxtaposition with extant CNN and transformer-based paradigms for nuanced ship categorization.
Databáze: Directory of Open Access Journals