├── .gitignore ├── Deraining ├── Datasets │ └── README.md ├── Options │ ├── Deraining_transMamba_Rain200H.yml │ └── Deraining_transMamba_Rain200L.yml ├── compute_psnr.py ├── pretrained_models │ └── README.md ├── scrpit.sh ├── test.py └── util.py ├── INSTALL.md ├── README.md ├── assets └── TransMamba.png ├── basicsr ├── __pycache__ │ └── version.cpython-39.pyc ├── data │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-38.pyc │ │ ├── __init__.cpython-39.pyc │ │ ├── data_sampler.cpython-38.pyc │ │ ├── data_sampler.cpython-39.pyc │ │ ├── data_util.cpython-38.pyc │ │ ├── data_util.cpython-39.pyc │ │ ├── ffhq_dataset.cpython-38.pyc │ │ ├── ffhq_dataset.cpython-39.pyc │ │ ├── paired_image_dataset.cpython-38.pyc │ │ ├── paired_image_dataset.cpython-39.pyc │ │ ├── prefetch_dataloader.cpython-38.pyc │ │ ├── prefetch_dataloader.cpython-39.pyc │ │ ├── reds_dataset.cpython-38.pyc │ │ ├── reds_dataset.cpython-39.pyc │ │ ├── single_image_dataset.cpython-38.pyc │ │ ├── single_image_dataset.cpython-39.pyc │ │ ├── transforms.cpython-38.pyc │ │ ├── transforms.cpython-39.pyc │ │ ├── video_test_dataset.cpython-38.pyc │ │ ├── video_test_dataset.cpython-39.pyc │ │ ├── vimeo90k_dataset.cpython-38.pyc │ │ └── vimeo90k_dataset.cpython-39.pyc │ ├── data_sampler.py │ ├── data_util.py │ ├── ffhq_dataset.py │ ├── meta_info │ │ ├── meta_info_DIV2K800sub_GT.txt │ │ ├── meta_info_REDS4_test_GT.txt │ │ ├── meta_info_REDS_GT.txt │ │ ├── meta_info_REDSofficial4_test_GT.txt │ │ ├── meta_info_REDSval_official_test_GT.txt │ │ ├── meta_info_Vimeo90K_test_GT.txt │ │ ├── meta_info_Vimeo90K_test_fast_GT.txt │ │ ├── meta_info_Vimeo90K_test_medium_GT.txt │ │ ├── meta_info_Vimeo90K_test_slow_GT.txt │ │ └── meta_info_Vimeo90K_train_GT.txt │ ├── paired_image_dataset.py │ ├── prefetch_dataloader.py │ ├── reds_dataset.py │ ├── single_image_dataset.py │ ├── transforms.py │ ├── video_test_dataset.py │ └── vimeo90k_dataset.py ├── metrics │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-38.pyc │ │ ├── __init__.cpython-39.pyc │ │ ├── metric_util.cpython-39.pyc │ │ ├── niqe.cpython-38.pyc │ │ ├── niqe.cpython-39.pyc │ │ ├── psnr_ssim.cpython-38.pyc │ │ └── psnr_ssim.cpython-39.pyc │ ├── fid.py │ ├── metric_util.py │ ├── niqe.py │ ├── niqe_pris_params.npz │ └── psnr_ssim.py ├── models │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-38.pyc │ │ ├── __init__.cpython-39.pyc │ │ ├── base_model.cpython-38.pyc │ │ ├── base_model.cpython-39.pyc │ │ ├── image_restoration_model.cpython-38.pyc │ │ ├── image_restoration_model.cpython-39.pyc │ │ ├── lr_scheduler.cpython-38.pyc │ │ └── lr_scheduler.cpython-39.pyc │ ├── archs │ │ ├── __init__.py │ │ ├── __pycache__ │ │ │ ├── __init__.cpython-38.pyc │ │ │ ├── __init__.cpython-39.pyc │ │ │ ├── derainmamba_arch.cpython-38.pyc │ │ │ ├── derainmamba_arch.cpython-39.pyc │ │ │ ├── fft_conv.cpython-38.pyc │ │ │ ├── fftformer_arch.cpython-38.pyc │ │ │ ├── fftformer_arch.cpython-39.pyc │ │ │ ├── mamba.cpython-39.pyc │ │ │ ├── mambair_arch.cpython-39.pyc │ │ │ ├── restormer_arch.cpython-38.pyc │ │ │ ├── restormer_arch.cpython-39.pyc │ │ │ ├── sbformer_arch.cpython-38.pyc │ │ │ ├── sbformer_arch.cpython-39.pyc │ │ │ └── transMamba_arch.cpython-39.pyc │ │ ├── arch_util.py │ │ ├── mamba.py │ │ └── transMamba_arch.py │ ├── base_model.py │ ├── image_restoration_model.py │ ├── losses │ │ ├── __init__.py │ │ ├── __pycache__ │ │ │ ├── __init__.cpython-38.pyc │ │ │ ├── __init__.cpython-39.pyc │ │ │ ├── loss_util.cpython-38.pyc │ │ │ ├── loss_util.cpython-39.pyc │ │ │ ├── losses.cpython-38.pyc │ │ │ └── losses.cpython-39.pyc │ │ ├── loss_util.py │ │ └── losses.py │ └── lr_scheduler.py ├── train.py ├── utils │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-38.pyc │ │ ├── __init__.cpython-39.pyc │ │ ├── create_lmdb.cpython-38.pyc │ │ ├── create_lmdb.cpython-39.pyc │ │ ├── dist_util.cpython-38.pyc │ │ ├── dist_util.cpython-39.pyc │ │ ├── file_client.cpython-38.pyc │ │ ├── file_client.cpython-39.pyc │ │ ├── flow_util.cpython-39.pyc │ │ ├── img_util.cpython-38.pyc │ │ ├── img_util.cpython-39.pyc │ │ ├── lmdb_util.cpython-38.pyc │ │ ├── lmdb_util.cpython-39.pyc │ │ ├── logger.cpython-38.pyc │ │ ├── logger.cpython-39.pyc │ │ ├── matlab_functions.cpython-39.pyc │ │ ├── misc.cpython-38.pyc │ │ ├── misc.cpython-39.pyc │ │ ├── options.cpython-38.pyc │ │ └── options.cpython-39.pyc │ ├── bundle_submissions.py │ ├── create_lmdb.py │ ├── dist_util.py │ ├── download_util.py │ ├── face_util.py │ ├── file_client.py │ ├── flow_util.py │ ├── img_util.py │ ├── lmdb_util.py │ ├── logger.py │ ├── matlab_functions.py │ ├── misc.py │ └── options.py └── version.py └── train.sh /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/.gitignore -------------------------------------------------------------------------------- /Deraining/Datasets/README.md: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Deraining/Options/Deraining_transMamba_Rain200H.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/Deraining/Options/Deraining_transMamba_Rain200H.yml -------------------------------------------------------------------------------- /Deraining/Options/Deraining_transMamba_Rain200L.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/Deraining/Options/Deraining_transMamba_Rain200L.yml -------------------------------------------------------------------------------- /Deraining/compute_psnr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/Deraining/compute_psnr.py -------------------------------------------------------------------------------- /Deraining/pretrained_models/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/Deraining/pretrained_models/README.md -------------------------------------------------------------------------------- /Deraining/scrpit.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/Deraining/scrpit.sh -------------------------------------------------------------------------------- /Deraining/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/Deraining/test.py -------------------------------------------------------------------------------- /Deraining/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/Deraining/util.py -------------------------------------------------------------------------------- /INSTALL.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/INSTALL.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/README.md -------------------------------------------------------------------------------- /assets/TransMamba.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/assets/TransMamba.png -------------------------------------------------------------------------------- /basicsr/__pycache__/version.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/__pycache__/version.cpython-39.pyc -------------------------------------------------------------------------------- /basicsr/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/data/__init__.py -------------------------------------------------------------------------------- /basicsr/data/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/data/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /basicsr/data/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/data/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /basicsr/data/__pycache__/data_sampler.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/data/__pycache__/data_sampler.cpython-38.pyc -------------------------------------------------------------------------------- /basicsr/data/__pycache__/data_sampler.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/data/__pycache__/data_sampler.cpython-39.pyc -------------------------------------------------------------------------------- /basicsr/data/__pycache__/data_util.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/data/__pycache__/data_util.cpython-38.pyc -------------------------------------------------------------------------------- /basicsr/data/__pycache__/data_util.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/data/__pycache__/data_util.cpython-39.pyc -------------------------------------------------------------------------------- /basicsr/data/__pycache__/ffhq_dataset.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/data/__pycache__/ffhq_dataset.cpython-38.pyc -------------------------------------------------------------------------------- /basicsr/data/__pycache__/ffhq_dataset.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/data/__pycache__/ffhq_dataset.cpython-39.pyc -------------------------------------------------------------------------------- /basicsr/data/__pycache__/paired_image_dataset.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/data/__pycache__/paired_image_dataset.cpython-38.pyc -------------------------------------------------------------------------------- /basicsr/data/__pycache__/paired_image_dataset.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/data/__pycache__/paired_image_dataset.cpython-39.pyc -------------------------------------------------------------------------------- /basicsr/data/__pycache__/prefetch_dataloader.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/data/__pycache__/prefetch_dataloader.cpython-38.pyc -------------------------------------------------------------------------------- /basicsr/data/__pycache__/prefetch_dataloader.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/data/__pycache__/prefetch_dataloader.cpython-39.pyc -------------------------------------------------------------------------------- /basicsr/data/__pycache__/reds_dataset.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/data/__pycache__/reds_dataset.cpython-38.pyc -------------------------------------------------------------------------------- /basicsr/data/__pycache__/reds_dataset.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/data/__pycache__/reds_dataset.cpython-39.pyc -------------------------------------------------------------------------------- /basicsr/data/__pycache__/single_image_dataset.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/data/__pycache__/single_image_dataset.cpython-38.pyc -------------------------------------------------------------------------------- /basicsr/data/__pycache__/single_image_dataset.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/data/__pycache__/single_image_dataset.cpython-39.pyc -------------------------------------------------------------------------------- /basicsr/data/__pycache__/transforms.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/data/__pycache__/transforms.cpython-38.pyc -------------------------------------------------------------------------------- /basicsr/data/__pycache__/transforms.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/data/__pycache__/transforms.cpython-39.pyc -------------------------------------------------------------------------------- /basicsr/data/__pycache__/video_test_dataset.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/data/__pycache__/video_test_dataset.cpython-38.pyc -------------------------------------------------------------------------------- /basicsr/data/__pycache__/video_test_dataset.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/data/__pycache__/video_test_dataset.cpython-39.pyc -------------------------------------------------------------------------------- /basicsr/data/__pycache__/vimeo90k_dataset.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/data/__pycache__/vimeo90k_dataset.cpython-38.pyc -------------------------------------------------------------------------------- /basicsr/data/__pycache__/vimeo90k_dataset.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/data/__pycache__/vimeo90k_dataset.cpython-39.pyc -------------------------------------------------------------------------------- /basicsr/data/data_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/data/data_sampler.py -------------------------------------------------------------------------------- /basicsr/data/data_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/data/data_util.py -------------------------------------------------------------------------------- /basicsr/data/ffhq_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/data/ffhq_dataset.py -------------------------------------------------------------------------------- /basicsr/data/meta_info/meta_info_DIV2K800sub_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/data/meta_info/meta_info_DIV2K800sub_GT.txt -------------------------------------------------------------------------------- /basicsr/data/meta_info/meta_info_REDS4_test_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/data/meta_info/meta_info_REDS4_test_GT.txt -------------------------------------------------------------------------------- /basicsr/data/meta_info/meta_info_REDS_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/data/meta_info/meta_info_REDS_GT.txt -------------------------------------------------------------------------------- /basicsr/data/meta_info/meta_info_REDSofficial4_test_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/data/meta_info/meta_info_REDSofficial4_test_GT.txt -------------------------------------------------------------------------------- /basicsr/data/meta_info/meta_info_REDSval_official_test_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/data/meta_info/meta_info_REDSval_official_test_GT.txt -------------------------------------------------------------------------------- /basicsr/data/meta_info/meta_info_Vimeo90K_test_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/data/meta_info/meta_info_Vimeo90K_test_GT.txt -------------------------------------------------------------------------------- /basicsr/data/meta_info/meta_info_Vimeo90K_test_fast_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/data/meta_info/meta_info_Vimeo90K_test_fast_GT.txt -------------------------------------------------------------------------------- /basicsr/data/meta_info/meta_info_Vimeo90K_test_medium_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/data/meta_info/meta_info_Vimeo90K_test_medium_GT.txt -------------------------------------------------------------------------------- /basicsr/data/meta_info/meta_info_Vimeo90K_test_slow_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/data/meta_info/meta_info_Vimeo90K_test_slow_GT.txt -------------------------------------------------------------------------------- /basicsr/data/meta_info/meta_info_Vimeo90K_train_GT.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/data/meta_info/meta_info_Vimeo90K_train_GT.txt -------------------------------------------------------------------------------- /basicsr/data/paired_image_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/data/paired_image_dataset.py -------------------------------------------------------------------------------- /basicsr/data/prefetch_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/data/prefetch_dataloader.py -------------------------------------------------------------------------------- /basicsr/data/reds_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/data/reds_dataset.py -------------------------------------------------------------------------------- /basicsr/data/single_image_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/data/single_image_dataset.py -------------------------------------------------------------------------------- /basicsr/data/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/data/transforms.py -------------------------------------------------------------------------------- /basicsr/data/video_test_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/data/video_test_dataset.py -------------------------------------------------------------------------------- /basicsr/data/vimeo90k_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/data/vimeo90k_dataset.py -------------------------------------------------------------------------------- /basicsr/metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/metrics/__init__.py -------------------------------------------------------------------------------- /basicsr/metrics/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/metrics/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /basicsr/metrics/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/metrics/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /basicsr/metrics/__pycache__/metric_util.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/metrics/__pycache__/metric_util.cpython-39.pyc -------------------------------------------------------------------------------- /basicsr/metrics/__pycache__/niqe.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/metrics/__pycache__/niqe.cpython-38.pyc -------------------------------------------------------------------------------- /basicsr/metrics/__pycache__/niqe.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/metrics/__pycache__/niqe.cpython-39.pyc -------------------------------------------------------------------------------- /basicsr/metrics/__pycache__/psnr_ssim.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/metrics/__pycache__/psnr_ssim.cpython-38.pyc -------------------------------------------------------------------------------- /basicsr/metrics/__pycache__/psnr_ssim.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/metrics/__pycache__/psnr_ssim.cpython-39.pyc -------------------------------------------------------------------------------- /basicsr/metrics/fid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/metrics/fid.py -------------------------------------------------------------------------------- /basicsr/metrics/metric_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/metrics/metric_util.py -------------------------------------------------------------------------------- /basicsr/metrics/niqe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/metrics/niqe.py -------------------------------------------------------------------------------- /basicsr/metrics/niqe_pris_params.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/metrics/niqe_pris_params.npz -------------------------------------------------------------------------------- /basicsr/metrics/psnr_ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/metrics/psnr_ssim.py -------------------------------------------------------------------------------- /basicsr/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/models/__init__.py -------------------------------------------------------------------------------- /basicsr/models/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/models/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /basicsr/models/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/models/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /basicsr/models/__pycache__/base_model.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/models/__pycache__/base_model.cpython-38.pyc -------------------------------------------------------------------------------- /basicsr/models/__pycache__/base_model.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/models/__pycache__/base_model.cpython-39.pyc -------------------------------------------------------------------------------- /basicsr/models/__pycache__/image_restoration_model.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/models/__pycache__/image_restoration_model.cpython-38.pyc -------------------------------------------------------------------------------- /basicsr/models/__pycache__/image_restoration_model.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/models/__pycache__/image_restoration_model.cpython-39.pyc -------------------------------------------------------------------------------- /basicsr/models/__pycache__/lr_scheduler.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/models/__pycache__/lr_scheduler.cpython-38.pyc -------------------------------------------------------------------------------- /basicsr/models/__pycache__/lr_scheduler.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/models/__pycache__/lr_scheduler.cpython-39.pyc -------------------------------------------------------------------------------- /basicsr/models/archs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/models/archs/__init__.py -------------------------------------------------------------------------------- /basicsr/models/archs/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/models/archs/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /basicsr/models/archs/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/models/archs/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /basicsr/models/archs/__pycache__/derainmamba_arch.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/models/archs/__pycache__/derainmamba_arch.cpython-38.pyc -------------------------------------------------------------------------------- /basicsr/models/archs/__pycache__/derainmamba_arch.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/models/archs/__pycache__/derainmamba_arch.cpython-39.pyc -------------------------------------------------------------------------------- /basicsr/models/archs/__pycache__/fft_conv.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/models/archs/__pycache__/fft_conv.cpython-38.pyc -------------------------------------------------------------------------------- /basicsr/models/archs/__pycache__/fftformer_arch.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/models/archs/__pycache__/fftformer_arch.cpython-38.pyc -------------------------------------------------------------------------------- /basicsr/models/archs/__pycache__/fftformer_arch.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/models/archs/__pycache__/fftformer_arch.cpython-39.pyc -------------------------------------------------------------------------------- /basicsr/models/archs/__pycache__/mamba.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/models/archs/__pycache__/mamba.cpython-39.pyc -------------------------------------------------------------------------------- /basicsr/models/archs/__pycache__/mambair_arch.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/models/archs/__pycache__/mambair_arch.cpython-39.pyc -------------------------------------------------------------------------------- /basicsr/models/archs/__pycache__/restormer_arch.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/models/archs/__pycache__/restormer_arch.cpython-38.pyc -------------------------------------------------------------------------------- /basicsr/models/archs/__pycache__/restormer_arch.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/models/archs/__pycache__/restormer_arch.cpython-39.pyc -------------------------------------------------------------------------------- /basicsr/models/archs/__pycache__/sbformer_arch.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/models/archs/__pycache__/sbformer_arch.cpython-38.pyc -------------------------------------------------------------------------------- /basicsr/models/archs/__pycache__/sbformer_arch.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/models/archs/__pycache__/sbformer_arch.cpython-39.pyc -------------------------------------------------------------------------------- /basicsr/models/archs/__pycache__/transMamba_arch.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/models/archs/__pycache__/transMamba_arch.cpython-39.pyc -------------------------------------------------------------------------------- /basicsr/models/archs/arch_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/models/archs/arch_util.py -------------------------------------------------------------------------------- /basicsr/models/archs/mamba.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/models/archs/mamba.py -------------------------------------------------------------------------------- /basicsr/models/archs/transMamba_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/models/archs/transMamba_arch.py -------------------------------------------------------------------------------- /basicsr/models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/models/base_model.py -------------------------------------------------------------------------------- /basicsr/models/image_restoration_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/models/image_restoration_model.py -------------------------------------------------------------------------------- /basicsr/models/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/models/losses/__init__.py -------------------------------------------------------------------------------- /basicsr/models/losses/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/models/losses/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /basicsr/models/losses/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/models/losses/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /basicsr/models/losses/__pycache__/loss_util.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/models/losses/__pycache__/loss_util.cpython-38.pyc -------------------------------------------------------------------------------- /basicsr/models/losses/__pycache__/loss_util.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/models/losses/__pycache__/loss_util.cpython-39.pyc -------------------------------------------------------------------------------- /basicsr/models/losses/__pycache__/losses.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/models/losses/__pycache__/losses.cpython-38.pyc -------------------------------------------------------------------------------- /basicsr/models/losses/__pycache__/losses.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/models/losses/__pycache__/losses.cpython-39.pyc -------------------------------------------------------------------------------- /basicsr/models/losses/loss_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/models/losses/loss_util.py -------------------------------------------------------------------------------- /basicsr/models/losses/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/models/losses/losses.py -------------------------------------------------------------------------------- /basicsr/models/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/models/lr_scheduler.py -------------------------------------------------------------------------------- /basicsr/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/train.py -------------------------------------------------------------------------------- /basicsr/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/utils/__init__.py -------------------------------------------------------------------------------- /basicsr/utils/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/utils/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /basicsr/utils/__pycache__/__init__.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/utils/__pycache__/__init__.cpython-39.pyc -------------------------------------------------------------------------------- /basicsr/utils/__pycache__/create_lmdb.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/utils/__pycache__/create_lmdb.cpython-38.pyc -------------------------------------------------------------------------------- /basicsr/utils/__pycache__/create_lmdb.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/utils/__pycache__/create_lmdb.cpython-39.pyc -------------------------------------------------------------------------------- /basicsr/utils/__pycache__/dist_util.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/utils/__pycache__/dist_util.cpython-38.pyc -------------------------------------------------------------------------------- /basicsr/utils/__pycache__/dist_util.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/utils/__pycache__/dist_util.cpython-39.pyc -------------------------------------------------------------------------------- /basicsr/utils/__pycache__/file_client.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/utils/__pycache__/file_client.cpython-38.pyc -------------------------------------------------------------------------------- /basicsr/utils/__pycache__/file_client.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/utils/__pycache__/file_client.cpython-39.pyc -------------------------------------------------------------------------------- /basicsr/utils/__pycache__/flow_util.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/utils/__pycache__/flow_util.cpython-39.pyc -------------------------------------------------------------------------------- /basicsr/utils/__pycache__/img_util.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/utils/__pycache__/img_util.cpython-38.pyc -------------------------------------------------------------------------------- /basicsr/utils/__pycache__/img_util.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/utils/__pycache__/img_util.cpython-39.pyc -------------------------------------------------------------------------------- /basicsr/utils/__pycache__/lmdb_util.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/utils/__pycache__/lmdb_util.cpython-38.pyc -------------------------------------------------------------------------------- /basicsr/utils/__pycache__/lmdb_util.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/utils/__pycache__/lmdb_util.cpython-39.pyc -------------------------------------------------------------------------------- /basicsr/utils/__pycache__/logger.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/utils/__pycache__/logger.cpython-38.pyc -------------------------------------------------------------------------------- /basicsr/utils/__pycache__/logger.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/utils/__pycache__/logger.cpython-39.pyc -------------------------------------------------------------------------------- /basicsr/utils/__pycache__/matlab_functions.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/utils/__pycache__/matlab_functions.cpython-39.pyc -------------------------------------------------------------------------------- /basicsr/utils/__pycache__/misc.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/utils/__pycache__/misc.cpython-38.pyc -------------------------------------------------------------------------------- /basicsr/utils/__pycache__/misc.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/utils/__pycache__/misc.cpython-39.pyc -------------------------------------------------------------------------------- /basicsr/utils/__pycache__/options.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/utils/__pycache__/options.cpython-38.pyc -------------------------------------------------------------------------------- /basicsr/utils/__pycache__/options.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/utils/__pycache__/options.cpython-39.pyc -------------------------------------------------------------------------------- /basicsr/utils/bundle_submissions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/utils/bundle_submissions.py -------------------------------------------------------------------------------- /basicsr/utils/create_lmdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/utils/create_lmdb.py -------------------------------------------------------------------------------- /basicsr/utils/dist_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/utils/dist_util.py -------------------------------------------------------------------------------- /basicsr/utils/download_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/utils/download_util.py -------------------------------------------------------------------------------- /basicsr/utils/face_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/utils/face_util.py -------------------------------------------------------------------------------- /basicsr/utils/file_client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/utils/file_client.py -------------------------------------------------------------------------------- /basicsr/utils/flow_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/utils/flow_util.py -------------------------------------------------------------------------------- /basicsr/utils/img_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/utils/img_util.py -------------------------------------------------------------------------------- /basicsr/utils/lmdb_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/utils/lmdb_util.py -------------------------------------------------------------------------------- /basicsr/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/utils/logger.py -------------------------------------------------------------------------------- /basicsr/utils/matlab_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/utils/matlab_functions.py -------------------------------------------------------------------------------- /basicsr/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/utils/misc.py -------------------------------------------------------------------------------- /basicsr/utils/options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/utils/options.py -------------------------------------------------------------------------------- /basicsr/version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/basicsr/version.py -------------------------------------------------------------------------------- /train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sunshangquan/TransMamba/HEAD/train.sh --------------------------------------------------------------------------------