├── LICENSE ├── README.md ├── data ├── Sandiego.mat └── abu-urban-1.mat ├── datasets ├── HADDatasets.py └── __init__.py ├── detect_any_HSI.py ├── losses ├── __init__.py └── msgms_loss.py ├── models ├── __init__.py ├── aetnet.py ├── spectralTransformer.py └── swin.py ├── requirements.txt ├── result └── 50bands_aviris_ng_1 │ └── seed60_zeromask_msgms4loss_dim32 │ └── best_model.pt ├── test.py ├── train.py └── utils ├── RX.py ├── __init__.py ├── gen_mask.py └── utils.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhaoxuLi123/AETNet/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhaoxuLi123/AETNet/HEAD/README.md -------------------------------------------------------------------------------- /data/Sandiego.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhaoxuLi123/AETNet/HEAD/data/Sandiego.mat -------------------------------------------------------------------------------- /data/abu-urban-1.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhaoxuLi123/AETNet/HEAD/data/abu-urban-1.mat -------------------------------------------------------------------------------- /datasets/HADDatasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhaoxuLi123/AETNet/HEAD/datasets/HADDatasets.py -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /detect_any_HSI.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhaoxuLi123/AETNet/HEAD/detect_any_HSI.py -------------------------------------------------------------------------------- /losses/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /losses/msgms_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhaoxuLi123/AETNet/HEAD/losses/msgms_loss.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/aetnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhaoxuLi123/AETNet/HEAD/models/aetnet.py -------------------------------------------------------------------------------- /models/spectralTransformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhaoxuLi123/AETNet/HEAD/models/spectralTransformer.py -------------------------------------------------------------------------------- /models/swin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhaoxuLi123/AETNet/HEAD/models/swin.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | torch 2 | kornia 3 | sklearn 4 | timm 5 | einops 6 | scikit-image -------------------------------------------------------------------------------- /result/50bands_aviris_ng_1/seed60_zeromask_msgms4loss_dim32/best_model.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhaoxuLi123/AETNet/HEAD/result/50bands_aviris_ng_1/seed60_zeromask_msgms4loss_dim32/best_model.pt -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhaoxuLi123/AETNet/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhaoxuLi123/AETNet/HEAD/train.py -------------------------------------------------------------------------------- /utils/RX.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhaoxuLi123/AETNet/HEAD/utils/RX.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhaoxuLi123/AETNet/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/gen_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhaoxuLi123/AETNet/HEAD/utils/gen_mask.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhaoxuLi123/AETNet/HEAD/utils/utils.py --------------------------------------------------------------------------------