gitextract_1m56b8y9/ ├── .gitignore ├── Deraining/ │ ├── Deraining/ │ │ ├── Datasets/ │ │ │ └── README.md │ │ ├── Metric/ │ │ │ └── PSNR.py │ │ ├── Options/ │ │ │ ├── Deraining_mamber28.yml │ │ │ ├── Deraining_mamber32.yml │ │ │ ├── Deraining_mamber33.yml │ │ │ └── Deraining_mamber34.yml │ │ ├── README.md │ │ ├── download_data.py │ │ ├── evaluate_PSNR_SSIM.m │ │ ├── test.py │ │ └── utils.py │ ├── Deraining_test.sh │ ├── Deraining_train.sh │ ├── INSTALL.md │ ├── LICENSE.md │ ├── VERSION │ ├── basicsr/ │ │ ├── data/ │ │ │ ├── __init__.py │ │ │ ├── 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 │ │ │ ├── fid.py │ │ │ ├── metric_util.py │ │ │ ├── niqe.py │ │ │ ├── niqe_pris_params.npz │ │ │ └── psnr_ssim.py │ │ ├── models/ │ │ │ ├── __init__.py │ │ │ ├── archs/ │ │ │ │ ├── __init__.py │ │ │ │ ├── arch_util.py │ │ │ │ ├── common.py │ │ │ │ ├── mamber32_arch.py │ │ │ │ ├── mamber33_arch.py │ │ │ │ └── restormer_arch.py │ │ │ ├── base_model.py │ │ │ ├── image_restoration_model.py │ │ │ ├── losses/ │ │ │ │ ├── __init__.py │ │ │ │ ├── loss_util.py │ │ │ │ └── losses.py │ │ │ └── lr_scheduler.py │ │ ├── test.py │ │ ├── test_deraining.py │ │ ├── train.py │ │ ├── utils/ │ │ │ ├── __init__.py │ │ │ ├── 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 │ │ ├── utils2.py │ │ └── version.py │ ├── pip.sh │ ├── setup.cfg │ ├── setup.py │ └── train.sh ├── Mamba/ │ ├── .gitignore │ └── kernels/ │ └── selective_scan/ │ ├── README.md │ ├── csrc/ │ │ └── selective_scan/ │ │ ├── cub_extra.cuh │ │ ├── cus/ │ │ │ ├── selective_scan.cpp │ │ │ ├── selective_scan_bwd_kernel.cuh │ │ │ ├── selective_scan_core_bwd.cu │ │ │ ├── selective_scan_core_fwd.cu │ │ │ └── selective_scan_fwd_kernel.cuh │ │ ├── cusndstate/ │ │ │ ├── selective_scan_bwd_kernel_ndstate.cuh │ │ │ ├── selective_scan_core_bwd.cu │ │ │ ├── selective_scan_core_fwd.cu │ │ │ ├── selective_scan_fwd_kernel_ndstate.cuh │ │ │ ├── selective_scan_ndstate.cpp │ │ │ └── selective_scan_ndstate.h │ │ ├── cusnrow/ │ │ │ ├── selective_scan_bwd_kernel_nrow.cuh │ │ │ ├── selective_scan_core_bwd.cu │ │ │ ├── selective_scan_core_bwd2.cu │ │ │ ├── selective_scan_core_bwd3.cu │ │ │ ├── selective_scan_core_bwd4.cu │ │ │ ├── selective_scan_core_fwd.cu │ │ │ ├── selective_scan_core_fwd2.cu │ │ │ ├── selective_scan_core_fwd3.cu │ │ │ ├── selective_scan_core_fwd4.cu │ │ │ ├── selective_scan_fwd_kernel_nrow.cuh │ │ │ └── selective_scan_nrow.cpp │ │ ├── cusoflex/ │ │ │ ├── selective_scan_bwd_kernel_oflex.cuh │ │ │ ├── selective_scan_core_bwd.cu │ │ │ ├── selective_scan_core_fwd.cu │ │ │ ├── selective_scan_fwd_kernel_oflex.cuh │ │ │ └── selective_scan_oflex.cpp │ │ ├── reverse_scan.cuh │ │ ├── selective_scan.h │ │ ├── selective_scan_common.h │ │ ├── static_switch.h │ │ └── uninitialized_copy.cuh │ ├── setup.py │ └── test_selective_scan.py ├── README.md ├── RealSR/ │ ├── .gitignore │ ├── Metric/ │ │ ├── DISTS/ │ │ │ ├── DISTS_pytorch/ │ │ │ │ ├── DISTS_pt.py │ │ │ │ └── weights.pt │ │ │ ├── DISTS_tensorflow/ │ │ │ │ └── DISTS_tf.py │ │ │ ├── LICENSE │ │ │ └── requirements.txt │ │ ├── LPIPS.py │ │ ├── PSNR.py │ │ ├── dists.py │ │ └── pip.sh │ ├── VERSION │ ├── VmambaIR/ │ │ ├── __init__.py │ │ ├── archs/ │ │ │ ├── MambaRealSR11_arch.py │ │ │ ├── __init__.py │ │ │ ├── attention.py │ │ │ ├── common.py │ │ │ ├── discriminator_arch.py │ │ │ └── srvgg_arch.py │ │ ├── data/ │ │ │ ├── __init__.py │ │ │ ├── data_util.py │ │ │ ├── deblur_paired_dataset.py │ │ │ ├── diffir_dataset.py │ │ │ ├── diffir_paired_dataset.py │ │ │ ├── gaussiandenoising_paired_dataset.py │ │ │ ├── realesrgan400_dataset.py │ │ │ ├── realesrgan_dataset.py │ │ │ ├── realesrgan_memery_dataset.py │ │ │ ├── realesrgan_paired_dataset.py │ │ │ └── transforms.py │ │ ├── losses/ │ │ │ ├── __init__.py │ │ │ └── my_loss.py │ │ ├── models/ │ │ │ ├── MambaRealSRGAN_model.py │ │ │ ├── MambaRealSRGANtest_model.py │ │ │ ├── MambaRealSR_model.py │ │ │ ├── __init__.py │ │ │ └── lr_scheduler.py │ │ ├── test.py │ │ ├── train.py │ │ ├── train_pipeline.py │ │ ├── utils/ │ │ │ ├── __init__.py │ │ │ ├── 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 │ │ ├── utils.py │ │ └── weights/ │ │ └── README.md │ ├── inference.py │ ├── ldm/ │ │ ├── classifier.py │ │ ├── lr_scheduler.py │ │ ├── util.py │ │ └── util2.py │ ├── metric.sh │ ├── options/ │ │ ├── mambaSR11GAN_x4.yml │ │ ├── mambaSR11_x4.yml │ │ ├── mambaSR11m_x4.yml │ │ └── test_mambaSR11GAN_x4.yml │ ├── pip.sh │ ├── requirements.txt │ ├── scripts/ │ │ ├── Metric/ │ │ │ ├── DISTS/ │ │ │ │ ├── DISTS_pytorch/ │ │ │ │ │ ├── DISTS_pt.py │ │ │ │ │ └── weights.pt │ │ │ │ ├── DISTS_tensorflow/ │ │ │ │ │ └── DISTS_tf.py │ │ │ │ ├── LICENSE │ │ │ │ └── requirements.txt │ │ │ ├── LPIPS.py │ │ │ ├── PSNR.py │ │ │ ├── dists.py │ │ │ └── pip.sh │ │ ├── extract_subimages.py │ │ ├── extract_subimages_DF2K.py │ │ ├── generate_meta_info.py │ │ ├── generate_meta_info_DF2K.py │ │ ├── generate_meta_info_DF2K_memo.py │ │ ├── generate_meta_info_OST.py │ │ ├── generate_meta_info_pairdata.py │ │ ├── generate_multiscale_DF2K.py │ │ ├── options/ │ │ │ ├── mambaSR11GAN_x4.yml │ │ │ ├── mambaSR11_x4.yml │ │ │ ├── mambaSR11m_x4.yml │ │ │ └── test_mambaSR11GAN_x4.yml │ │ └── pytorch2onnx.py │ ├── setup.cfg │ ├── setup.py │ ├── test.sh │ ├── tests/ │ │ ├── data/ │ │ │ ├── gt.lmdb/ │ │ │ │ ├── data.mdb │ │ │ │ ├── lock.mdb │ │ │ │ └── meta_info.txt │ │ │ ├── lq.lmdb/ │ │ │ │ ├── data.mdb │ │ │ │ ├── lock.mdb │ │ │ │ └── meta_info.txt │ │ │ ├── meta_info_gt.txt │ │ │ ├── meta_info_pair.txt │ │ │ ├── test_realesrgan_dataset.yml │ │ │ ├── test_realesrgan_model.yml │ │ │ ├── test_realesrgan_paired_dataset.yml │ │ │ └── test_realesrnet_model.yml │ │ ├── test_dataset.py │ │ ├── test_discriminator_arch.py │ │ ├── test_model.py │ │ └── test_utils.py │ ├── train_S1.sh │ └── train_S2.sh ├── SRGAN/ │ ├── .gitignore │ ├── Metric/ │ │ ├── DISTS/ │ │ │ ├── DISTS_pytorch/ │ │ │ │ ├── DISTS_pt.py │ │ │ │ └── weights.pt │ │ │ ├── DISTS_tensorflow/ │ │ │ │ └── DISTS_tf.py │ │ │ ├── LICENSE │ │ │ └── requirements.txt │ │ ├── LPIPS.py │ │ ├── PSNR.py │ │ ├── dists.py │ │ └── pip.sh │ ├── VERSION │ ├── VmambaIR/ │ │ ├── __init__.py │ │ ├── archs/ │ │ │ ├── MambaSISR6_arch.py │ │ │ ├── __init__.py │ │ │ ├── attention.py │ │ │ ├── common.py │ │ │ ├── discriminator_arch.py │ │ │ └── srvgg_arch.py │ │ ├── data/ │ │ │ ├── __init__.py │ │ │ ├── data_util.py │ │ │ ├── deblur_paired_dataset.py │ │ │ ├── diffir_dataset.py │ │ │ ├── diffir_paired_dataset.py │ │ │ ├── gaussiandenoising_paired_dataset.py │ │ │ └── transforms.py │ │ ├── losses/ │ │ │ ├── __init__.py │ │ │ └── my_loss.py │ │ ├── models/ │ │ │ ├── MambaSISR2_model.py │ │ │ ├── MambaSISRGAN_model.py │ │ │ ├── MambaSISR_model.py │ │ │ ├── __init__.py │ │ │ └── lr_scheduler.py │ │ ├── test.py │ │ ├── train.py │ │ ├── train_pipeline.py │ │ ├── utils/ │ │ │ ├── __init__.py │ │ │ ├── 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 │ │ ├── utils.py │ │ └── weights/ │ │ └── README.md │ ├── ldm/ │ │ ├── classifier.py │ │ ├── lr_scheduler.py │ │ ├── util.py │ │ └── util2.py │ ├── metric.sh │ ├── options/ │ │ ├── MambaSISR15GAN_x4.yml │ │ ├── MambaSISR15_x4.yml │ │ └── test_mamba15_x4.yml │ ├── pip.sh │ ├── requirements.txt │ ├── scripts/ │ │ ├── extract_subimages.py │ │ ├── extract_subimages_DF2K.py │ │ ├── generate_meta_info.py │ │ ├── generate_meta_info_DF2K.py │ │ ├── generate_meta_info_OST.py │ │ ├── generate_meta_info_pairdata.py │ │ ├── generate_multiscale_DF2K.py │ │ └── pytorch2onnx.py │ ├── setup.cfg │ ├── setup.py │ ├── test.sh │ ├── train_S1.sh │ └── train_S2.sh ├── install.md └── requirements.txt