Object Detection in Hyperspectral Image via Unified Spectral–Spatial Feature Aggregation

Deep learning-based hyperspectral image (HSI) classification and object detection techniques have gained significant attention due to their vital role in image content analysis, interpretation, and wider HSI applications. However, current hyperspectral object detection approaches predominantly emphasize either spectral or spatial information, overlooking the valuable complementary relationship between these two aspects. In this study, we present a novel \textbf{S}pectral-\textbf{S}patial \textbf{A}ggregation (S2ADet) object detector that effectively harnesses the rich spectral and spatial complementary information inherent in hyperspectral images. S2ADet comprises a hyperspectral information decoupling (HID) module, a two-stream feature extraction network, and a one-stage detection head. The HID module processes hyperspectral images by aggregating spectral and spatial information via band selection and principal components analysis, consequently reducing redundancy. Based on the acquired spatial and spectral aggregation information, we propose a feature aggregation two-stream network for interacting spectral-spatial features. Furthermore, to address the limitations of existing databases, we annotate an extensive dataset, designated as HOD3K, containing 3,242 hyperspectral images captured across diverse real-world scenes and encompassing three object classes. These images possess a resolution of 512x256 pixels and cover 16 bands ranging from 470 nm to 620 nm. Comprehensive experiments on two datasets demonstrate that S2ADet surpasses existing state-of-the-art methods, achieving robust and reliable results. The demo code and dataset of this work are publicly available at \url{https://github.com/hexiao-cs/S2ADet}.

[1]  Leyuan Fang,et al.  Querying Labeled for Unlabeled: Cross-Image Semantic Consistency Guided Semi-Supervised Semantic Segmentation , 2023, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[2]  Y. Zhang,et al.  Scene Representation in Bird’s-Eye View from Surrounding Cameras with Transformers , 2022, 2022 IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops (CVPRW).

[3]  L. Gool,et al.  Mask-guided Spectral-wise Transformer for Efficient Hyperspectral Image Reconstruction , 2021, 2022 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR).

[4]  L. Jiao,et al.  ViT-YOLO:Transformer-Based YOLO for Object Detection , 2021, 2021 IEEE/CVF International Conference on Computer Vision Workshops (ICCVW).

[5]  Huchuan Lu,et al.  Dynamic Context-Sensitive Filtering Network for Video Salient Object Detection , 2021, 2021 IEEE/CVF International Conference on Computer Vision (ICCV).

[6]  Weilin Huang,et al.  TOOD: Task-aligned One-stage Object Detection , 2021, 2021 IEEE/CVF International Conference on Computer Vision (ICCV).

[7]  D. Skočaj,et al.  DRÆM – A discriminatively trained reconstruction embedding for surface anomaly detection , 2021, 2021 IEEE/CVF International Conference on Computer Vision (ICCV).

[8]  Kai Han,et al.  CMT: Convolutional Neural Networks Meet Vision Transformers , 2021, 2022 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR).

[9]  Lianru Gao,et al.  SpectralFormer: Rethinking Hyperspectral Image Classification With Transformers , 2021, IEEE Transactions on Geoscience and Remote Sensing.

[10]  Haoyang Yu,et al.  Target-Constrained Interference-Minimized Band Selection for Hyperspectral Target Detection , 2021, IEEE Transactions on Geoscience and Remote Sensing.

[11]  Chein-I. Chang,et al.  Hyperspectral Anomaly Detection: A Dual Theory of Hyperspectral Target Detection , 2021, IEEE Transactions on Geoscience and Remote Sensing.

[12]  T. Fung,et al.  Mapping multi-layered mangroves from multispectral, hyperspectral, and LiDAR data , 2021 .

[13]  Deyu Meng,et al.  Learning An Explicit Weighting Scheme for Adapting Complex HSI Noise , 2021, 2021 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR).

[14]  Lizhi Wang,et al.  Learning Tensor Low-Rank Prior for Hyperspectral Image Reconstruction , 2021, 2021 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR).

[15]  Chein-I Chang,et al.  An Effective Evaluation Tool for Hyperspectral Target Detection: 3D Receiver Operating Characteristic Curve Analysis , 2021, IEEE Transactions on Geoscience and Remote Sensing.

[16]  W. Shi,et al.  Asymmetric Weighted Logistic Metric Learning for Hyperspectral Target Detection , 2021, IEEE Transactions on Cybernetics.

[17]  Yanfei Zhong,et al.  An Anchor-Free Siamese Target Tracking Network for Hyperspectral Video , 2021, 2021 11th Workshop on Hyperspectral Imaging and Signal Processing: Evolution in Remote Sensing (WHISPERS).

[18]  Xiangyu Zhang,et al.  You Only Look One-level Feature , 2021, 2021 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR).

[19]  Qi Wang,et al.  Hyperspectral Band Selection via Optimal Neighborhood Reconstruction , 2020, IEEE Transactions on Geoscience and Remote Sensing.

[20]  Jing Bai,et al.  Class Incremental Learning With Few-Shots Based on Linear Programming for Hyperspectral Image Classification , 2020, IEEE Transactions on Cybernetics.

[21]  Bin Li,et al.  Deformable DETR: Deformable Transformers for End-to-End Object Detection , 2020, ICLR.

[22]  Liangpei Zhang,et al.  Few-Shot Hyperspectral Image Classification With Unknown Classes Using Multitask Deep Learning , 2020, IEEE Transactions on Geoscience and Remote Sensing.

[23]  Matteo Matteucci,et al.  Deep Learning for Land Use and Land Cover Classification Based on Hyperspectral and Multispectral Earth Observation Data: A Review , 2020, Remote. Sens..

[24]  Nicolas Usunier,et al.  End-to-End Object Detection with Transformers , 2020, ECCV.

[25]  Xiao Xiang Zhu,et al.  Nonlocal Graph Convolutional Networks for Hyperspectral Image Classification , 2020, IEEE Transactions on Geoscience and Remote Sensing.

[26]  Qingshan Liu,et al.  Classification of Hyperspectral and LiDAR Data Using Coupled CNNs , 2020, IEEE Transactions on Geoscience and Remote Sensing.

[27]  Naoto Yokoya,et al.  Learning-Shared Cross-Modality Representation Using Multispectral-LiDAR and Hyperspectral Data , 2019, IEEE Geoscience and Remote Sensing Letters.

[28]  Dragomir Anguelov,et al.  Scalability in Perception for Autonomous Driving: Waymo Open Dataset , 2019, 2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR).

[29]  Alper Koz,et al.  Ground-Based Hyperspectral Image Surveillance Systems for Explosive Detection: Part I—State of the Art and Challenges , 2019, IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing.

[30]  V. Athitsos,et al.  lambda-Net: Reconstruct Hyperspectral Images From a Snapshot Measurement , 2019, 2019 IEEE/CVF International Conference on Computer Vision (ICCV).

[31]  Yuning Jiang,et al.  FoveaBox: Beyound Anchor-Based Object Detection , 2019, IEEE Transactions on Image Processing.

[32]  Ying Fu,et al.  Hyperspectral Image Reconstruction Using a Deep Spatial-Spectral Prior , 2019, 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR).

[33]  Jon Atli Benediktsson,et al.  Deep Learning for Hyperspectral Image Classification: An Overview , 2019, IEEE Transactions on Geoscience and Remote Sensing.

[34]  Y. Fu,et al.  Rethinking Classification and Localization for Object Detection , 2019, 2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR).

[35]  Huajun Feng,et al.  Libra R-CNN: Towards Balanced Learning for Object Detection , 2019, 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR).

[36]  Hao Chen,et al.  FCOS: Fully Convolutional One-Stage Object Detection , 2019, 2019 IEEE/CVF International Conference on Computer Vision (ICCV).

[37]  Bo Du,et al.  Binary-Class Collaborative Representation for Target Detection in Hyperspectral Images , 2019, IEEE Geoscience and Remote Sensing Letters.

[38]  Jun Zhou,et al.  Material Based Object Tracking in Hyperspectral Videos , 2018, IEEE Transactions on Image Processing.

[39]  Matti Pietikäinen,et al.  Deep Learning for Generic Object Detection: A Survey , 2018, International Journal of Computer Vision.

[40]  In-So Kweon,et al.  CBAM: Convolutional Block Attention Module , 2018, ECCV.

[41]  Xindong Wu,et al.  Object Detection With Deep Learning: A Review , 2018, IEEE Transactions on Neural Networks and Learning Systems.

[42]  Jiebo Luo,et al.  DOTA: A Large-Scale Dataset for Object Detection in Aerial Images , 2017, 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition.

[43]  Kaiming He,et al.  Focal Loss for Dense Object Detection , 2017, 2017 IEEE International Conference on Computer Vision (ICCV).

[44]  Yanfeng Gu,et al.  Tensor Matched Subspace Detector for Hyperspectral Target Detection , 2017, IEEE Transactions on Geoscience and Remote Sensing.

[45]  Ross B. Girshick,et al.  Mask R-CNN , 2017, 1703.06870.

[46]  Sen Jia,et al.  Convolutional neural networks for hyperspectral image classification , 2017, Neurocomputing.

[47]  Ali Farhadi,et al.  YOLO9000: Better, Faster, Stronger , 2016, 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[48]  Serge J. Belongie,et al.  Feature Pyramid Networks for Object Detection , 2016, 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[49]  Wei Liu,et al.  SSD: Single Shot MultiBox Detector , 2015, ECCV.

[50]  Kaiming He,et al.  Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks , 2015, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[51]  Ross B. Girshick,et al.  Fast R-CNN , 2015, 1504.08083.

[52]  Trevor Darrell,et al.  Rich Feature Hierarchies for Accurate Object Detection and Semantic Segmentation , 2013, 2014 IEEE Conference on Computer Vision and Pattern Recognition.

[53]  Antonio Torralba,et al.  LabelMe: Online Image Annotation and Applications , 2010, Proceedings of the IEEE.

[54]  M. Eklund,et al.  PEER REVIEWED PAPERS , 2003 .

[55]  Melba M. Crawford,et al.  Mapping of coastal wetlands via hyperspectral AVIRIS data , 1998, IGARSS '98. Sensing and Managing the Environment. 1998 IEEE International Geoscience and Remote Sensing. Symposium Proceedings. (Cat. No.98CH36174).

[56]  Ronald G. Resmini,et al.  Mineral mapping with HYperspectral Digital Imagery Collection Experiment (HYDICE) sensor data at Cuprite, Nevada, U.S.A. , 1997 .

[57]  Leyuan Fang,et al.  Hyperspectral Image Instance Segmentation Using Spectral–Spatial Feature Pyramid Network , 2023, IEEE Transactions on Geoscience and Remote Sensing.

[58]  A. Sharifi,et al.  Crop Type Classification by DESIS Hyperspectral Imagery and Machine Learning Algorithms , 2023, IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing.

[59]  Lizhe Wang,et al.  Lightweight Remote Sensing Change Detection With Progressive Feature Aggregation and Supervised Attention , 2023, IEEE Transactions on Geoscience and Remote Sensing.

[60]  Leyuan Fang,et al.  Toward the Vectorization of Hyperspectral Imagery , 2023, IEEE Transactions on Geoscience and Remote Sensing.

[61]  L. Jiao,et al.  Hyperspectral Imagery Classification Based on Contrastive Learning , 2022, IEEE Transactions on Geoscience and Remote Sensing.

[62]  Jie Chen,et al.  Object Detection in Hyperspectral Images , 2021, IEEE Signal Processing Letters.

[63]  Xudong Sun,et al.  Constrained-Target Band Selection With Subspace Partition for Hyperspectral Target Detection , 2021, IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing.

[64]  Meiping Song,et al.  Orthogonal Subspace Projection Target Detector for Hyperspectral Anomaly Detection , 2021, IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing.

[65]  Fang Miao,et al.  A Hyperspectral Image Classification Method Using Multifeature Vectors and Optimized KELM , 2021, IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing.

[66]  Heng Tao Shen,et al.  Principal Component Analysis , 2009, Encyclopedia of Biometrics.

[67]  S. Tabbone,et al.  Deep neural networks-based relevant latent representation learning for hyperspectral image classification , 2022, Pattern Recognit..