├── .gitignore ├── .readthedocs.yaml ├── LICENSE ├── README.md ├── benchmark ├── README.md └── datasets │ ├── CAVE │ ├── cave_build_lmdb.py │ ├── cave_build_test.py │ ├── cave_meta.txt │ ├── cave_preprocess.py │ ├── cave_test.txt │ └── cave_train.txt │ ├── Harvard │ ├── harvard_build_lmdb.py │ ├── harvard_build_test.py │ ├── harvard_meta.txt │ ├── harvard_test.txt │ └── harvard_train.txt │ ├── ICVL │ └── todo.txt │ └── RealHSI │ ├── build_test.py │ ├── real_build_lmdb.py │ ├── real_test.txt │ └── real_train.txt ├── docs ├── Makefile ├── make.bat ├── requirements.txt └── source │ ├── benchmark.md │ ├── conf.py │ ├── dataset.md │ ├── getstart.md │ └── index.md ├── hsiboard ├── app.py ├── box.py ├── cmp.py ├── main.py ├── table.py ├── util.py └── viewer.py ├── hsir ├── __init__.py ├── data │ ├── __init__.py │ ├── dataloader.py │ ├── dataset.py │ ├── transform │ │ ├── __init__.py │ │ ├── general.py │ │ ├── noise.py │ │ └── sr.py │ └── utils.py ├── model │ ├── __init__.py │ ├── denet.py │ ├── grunet.py │ ├── hsdt │ │ ├── __init__.py │ │ ├── arch.py │ │ ├── attention.py │ │ └── sepconv.py │ ├── hsidcnn.py │ ├── man │ │ ├── __init__.py │ │ ├── v1.py │ │ └── v2.py │ ├── memnet.py │ ├── qrnn3d │ │ ├── __init__.py │ │ ├── conv.py │ │ ├── net.py │ │ └── qrnn.py │ ├── restormer.py │ ├── sst.py │ ├── swinir.py │ ├── t3sc │ │ ├── __init__.py │ │ ├── base.py │ │ ├── configs │ │ │ └── t3sc.yaml │ │ ├── layers │ │ │ ├── __init__.py │ │ │ ├── encoding_layer.py │ │ │ ├── lowrank_sc_layer.py │ │ │ └── soft_thresholding.py │ │ ├── metrics │ │ │ ├── __init__.py │ │ │ ├── metrics.py │ │ │ └── utils.py │ │ ├── multilayer.py │ │ └── utils │ │ │ ├── __init__.py │ │ │ └── patches_handler.py │ ├── trq3d │ │ ├── __init__.py │ │ ├── attention.py │ │ ├── combinations.py │ │ ├── conv.py │ │ └── trq3d.py │ ├── uformer.py │ └── unet3d.py ├── schedule.py ├── scheduler.py └── trainer.py ├── hsirun ├── benchmark.py ├── test.py ├── train.py └── train_ssr.py └── setup.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/.gitignore -------------------------------------------------------------------------------- /.readthedocs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/.readthedocs.yaml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/README.md -------------------------------------------------------------------------------- /benchmark/README.md: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /benchmark/datasets/CAVE/cave_build_lmdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/benchmark/datasets/CAVE/cave_build_lmdb.py -------------------------------------------------------------------------------- /benchmark/datasets/CAVE/cave_build_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/benchmark/datasets/CAVE/cave_build_test.py -------------------------------------------------------------------------------- /benchmark/datasets/CAVE/cave_meta.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/benchmark/datasets/CAVE/cave_meta.txt -------------------------------------------------------------------------------- /benchmark/datasets/CAVE/cave_preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/benchmark/datasets/CAVE/cave_preprocess.py -------------------------------------------------------------------------------- /benchmark/datasets/CAVE/cave_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/benchmark/datasets/CAVE/cave_test.txt -------------------------------------------------------------------------------- /benchmark/datasets/CAVE/cave_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/benchmark/datasets/CAVE/cave_train.txt -------------------------------------------------------------------------------- /benchmark/datasets/Harvard/harvard_build_lmdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/benchmark/datasets/Harvard/harvard_build_lmdb.py -------------------------------------------------------------------------------- /benchmark/datasets/Harvard/harvard_build_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/benchmark/datasets/Harvard/harvard_build_test.py -------------------------------------------------------------------------------- /benchmark/datasets/Harvard/harvard_meta.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/benchmark/datasets/Harvard/harvard_meta.txt -------------------------------------------------------------------------------- /benchmark/datasets/Harvard/harvard_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/benchmark/datasets/Harvard/harvard_test.txt -------------------------------------------------------------------------------- /benchmark/datasets/Harvard/harvard_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/benchmark/datasets/Harvard/harvard_train.txt -------------------------------------------------------------------------------- /benchmark/datasets/ICVL/todo.txt: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /benchmark/datasets/RealHSI/build_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/benchmark/datasets/RealHSI/build_test.py -------------------------------------------------------------------------------- /benchmark/datasets/RealHSI/real_build_lmdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/benchmark/datasets/RealHSI/real_build_lmdb.py -------------------------------------------------------------------------------- /benchmark/datasets/RealHSI/real_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/benchmark/datasets/RealHSI/real_test.txt -------------------------------------------------------------------------------- /benchmark/datasets/RealHSI/real_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/benchmark/datasets/RealHSI/real_train.txt -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/make.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/docs/make.bat -------------------------------------------------------------------------------- /docs/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/docs/requirements.txt -------------------------------------------------------------------------------- /docs/source/benchmark.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/docs/source/benchmark.md -------------------------------------------------------------------------------- /docs/source/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/docs/source/conf.py -------------------------------------------------------------------------------- /docs/source/dataset.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/docs/source/dataset.md -------------------------------------------------------------------------------- /docs/source/getstart.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/docs/source/getstart.md -------------------------------------------------------------------------------- /docs/source/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/docs/source/index.md -------------------------------------------------------------------------------- /hsiboard/app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsiboard/app.py -------------------------------------------------------------------------------- /hsiboard/box.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsiboard/box.py -------------------------------------------------------------------------------- /hsiboard/cmp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsiboard/cmp.py -------------------------------------------------------------------------------- /hsiboard/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsiboard/main.py -------------------------------------------------------------------------------- /hsiboard/table.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsiboard/table.py -------------------------------------------------------------------------------- /hsiboard/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsiboard/util.py -------------------------------------------------------------------------------- /hsiboard/viewer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsiboard/viewer.py -------------------------------------------------------------------------------- /hsir/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/__init__.py -------------------------------------------------------------------------------- /hsir/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/data/__init__.py -------------------------------------------------------------------------------- /hsir/data/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/data/dataloader.py -------------------------------------------------------------------------------- /hsir/data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/data/dataset.py -------------------------------------------------------------------------------- /hsir/data/transform/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /hsir/data/transform/general.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/data/transform/general.py -------------------------------------------------------------------------------- /hsir/data/transform/noise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/data/transform/noise.py -------------------------------------------------------------------------------- /hsir/data/transform/sr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/data/transform/sr.py -------------------------------------------------------------------------------- /hsir/data/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/data/utils.py -------------------------------------------------------------------------------- /hsir/model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/model/__init__.py -------------------------------------------------------------------------------- /hsir/model/denet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/model/denet.py -------------------------------------------------------------------------------- /hsir/model/grunet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/model/grunet.py -------------------------------------------------------------------------------- /hsir/model/hsdt/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/model/hsdt/__init__.py -------------------------------------------------------------------------------- /hsir/model/hsdt/arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/model/hsdt/arch.py -------------------------------------------------------------------------------- /hsir/model/hsdt/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/model/hsdt/attention.py -------------------------------------------------------------------------------- /hsir/model/hsdt/sepconv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/model/hsdt/sepconv.py -------------------------------------------------------------------------------- /hsir/model/hsidcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/model/hsidcnn.py -------------------------------------------------------------------------------- /hsir/model/man/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /hsir/model/man/v1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/model/man/v1.py -------------------------------------------------------------------------------- /hsir/model/man/v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/model/man/v2.py -------------------------------------------------------------------------------- /hsir/model/memnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/model/memnet.py -------------------------------------------------------------------------------- /hsir/model/qrnn3d/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/model/qrnn3d/__init__.py -------------------------------------------------------------------------------- /hsir/model/qrnn3d/conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/model/qrnn3d/conv.py -------------------------------------------------------------------------------- /hsir/model/qrnn3d/net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/model/qrnn3d/net.py -------------------------------------------------------------------------------- /hsir/model/qrnn3d/qrnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/model/qrnn3d/qrnn.py -------------------------------------------------------------------------------- /hsir/model/restormer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/model/restormer.py -------------------------------------------------------------------------------- /hsir/model/sst.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/model/sst.py -------------------------------------------------------------------------------- /hsir/model/swinir.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/model/swinir.py -------------------------------------------------------------------------------- /hsir/model/t3sc/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/model/t3sc/__init__.py -------------------------------------------------------------------------------- /hsir/model/t3sc/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/model/t3sc/base.py -------------------------------------------------------------------------------- /hsir/model/t3sc/configs/t3sc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/model/t3sc/configs/t3sc.yaml -------------------------------------------------------------------------------- /hsir/model/t3sc/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/model/t3sc/layers/__init__.py -------------------------------------------------------------------------------- /hsir/model/t3sc/layers/encoding_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/model/t3sc/layers/encoding_layer.py -------------------------------------------------------------------------------- /hsir/model/t3sc/layers/lowrank_sc_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/model/t3sc/layers/lowrank_sc_layer.py -------------------------------------------------------------------------------- /hsir/model/t3sc/layers/soft_thresholding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/model/t3sc/layers/soft_thresholding.py -------------------------------------------------------------------------------- /hsir/model/t3sc/metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/model/t3sc/metrics/__init__.py -------------------------------------------------------------------------------- /hsir/model/t3sc/metrics/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/model/t3sc/metrics/metrics.py -------------------------------------------------------------------------------- /hsir/model/t3sc/metrics/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/model/t3sc/metrics/utils.py -------------------------------------------------------------------------------- /hsir/model/t3sc/multilayer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/model/t3sc/multilayer.py -------------------------------------------------------------------------------- /hsir/model/t3sc/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/model/t3sc/utils/__init__.py -------------------------------------------------------------------------------- /hsir/model/t3sc/utils/patches_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/model/t3sc/utils/patches_handler.py -------------------------------------------------------------------------------- /hsir/model/trq3d/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/model/trq3d/__init__.py -------------------------------------------------------------------------------- /hsir/model/trq3d/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/model/trq3d/attention.py -------------------------------------------------------------------------------- /hsir/model/trq3d/combinations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/model/trq3d/combinations.py -------------------------------------------------------------------------------- /hsir/model/trq3d/conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/model/trq3d/conv.py -------------------------------------------------------------------------------- /hsir/model/trq3d/trq3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/model/trq3d/trq3d.py -------------------------------------------------------------------------------- /hsir/model/uformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/model/uformer.py -------------------------------------------------------------------------------- /hsir/model/unet3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/model/unet3d.py -------------------------------------------------------------------------------- /hsir/schedule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/schedule.py -------------------------------------------------------------------------------- /hsir/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/scheduler.py -------------------------------------------------------------------------------- /hsir/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsir/trainer.py -------------------------------------------------------------------------------- /hsirun/benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsirun/benchmark.py -------------------------------------------------------------------------------- /hsirun/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsirun/test.py -------------------------------------------------------------------------------- /hsirun/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsirun/train.py -------------------------------------------------------------------------------- /hsirun/train_ssr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/hsirun/train_ssr.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bit-isp/HSIR/HEAD/setup.py --------------------------------------------------------------------------------