├── .idea ├── .gitignore ├── WF-Diff.iml ├── inspectionProfiles │ └── profiles_settings.xml ├── modules.xml └── vcs.xml ├── README.md ├── basicsr ├── __init__.py ├── archs │ ├── Padiff_arch │ │ ├── cfc_arch.py │ │ ├── dft_arch.py │ │ ├── diffx2_arch.py │ │ ├── unet_block │ │ │ ├── ConditionNet.py │ │ │ ├── cross_attention_module.py │ │ │ ├── diffusion.py │ │ │ ├── style_transfer.py │ │ │ ├── trans_block_eca.py │ │ │ └── unet.py │ │ ├── unetx2_arch.py │ │ └── wavelet.py │ ├── __init__.py │ ├── arch_util.py │ └── wfdiffx2_arch.py ├── data │ ├── MambaIR_paired_image_dataset.py │ ├── __init__.py │ ├── data_sampler.py │ ├── data_util.py │ ├── paired_image_dataset copy.py │ ├── paired_image_dataset.py │ ├── prefetch_dataloader.py │ ├── single_image_dataset.py │ └── transforms.py ├── losses │ ├── __init__.py │ ├── loss_util.py │ └── losses.py ├── metrics │ ├── __init__.py │ ├── metric_util.py │ └── psnr_ssim.py ├── models │ ├── Wfdiff_model.py │ ├── __init__.py │ ├── base_model.py │ ├── lr_scheduler.py │ ├── sr_model.py │ └── wavelet.py ├── test.py ├── train.py ├── utils │ ├── __init__.py │ ├── diff_util.py │ ├── dist_util.py │ ├── file_client.py │ ├── img_util.py │ ├── logger.py │ ├── matlab_functions.py │ ├── misc.py │ ├── options.py │ └── registry.py └── version.py ├── options ├── test │ └── test_wfdiff.yml └── train │ └── train_Wfdiff.yml └── setup.py /.idea/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/.idea/.gitignore -------------------------------------------------------------------------------- /.idea/WF-Diff.iml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/.idea/WF-Diff.iml -------------------------------------------------------------------------------- /.idea/inspectionProfiles/profiles_settings.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/.idea/inspectionProfiles/profiles_settings.xml -------------------------------------------------------------------------------- /.idea/modules.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/.idea/modules.xml -------------------------------------------------------------------------------- /.idea/vcs.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/.idea/vcs.xml -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/README.md -------------------------------------------------------------------------------- /basicsr/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/__init__.py -------------------------------------------------------------------------------- /basicsr/archs/Padiff_arch/cfc_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/archs/Padiff_arch/cfc_arch.py -------------------------------------------------------------------------------- /basicsr/archs/Padiff_arch/dft_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/archs/Padiff_arch/dft_arch.py -------------------------------------------------------------------------------- /basicsr/archs/Padiff_arch/diffx2_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/archs/Padiff_arch/diffx2_arch.py -------------------------------------------------------------------------------- /basicsr/archs/Padiff_arch/unet_block/ConditionNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/archs/Padiff_arch/unet_block/ConditionNet.py -------------------------------------------------------------------------------- /basicsr/archs/Padiff_arch/unet_block/cross_attention_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/archs/Padiff_arch/unet_block/cross_attention_module.py -------------------------------------------------------------------------------- /basicsr/archs/Padiff_arch/unet_block/diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/archs/Padiff_arch/unet_block/diffusion.py -------------------------------------------------------------------------------- /basicsr/archs/Padiff_arch/unet_block/style_transfer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/archs/Padiff_arch/unet_block/style_transfer.py -------------------------------------------------------------------------------- /basicsr/archs/Padiff_arch/unet_block/trans_block_eca.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/archs/Padiff_arch/unet_block/trans_block_eca.py -------------------------------------------------------------------------------- /basicsr/archs/Padiff_arch/unet_block/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/archs/Padiff_arch/unet_block/unet.py -------------------------------------------------------------------------------- /basicsr/archs/Padiff_arch/unetx2_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/archs/Padiff_arch/unetx2_arch.py -------------------------------------------------------------------------------- /basicsr/archs/Padiff_arch/wavelet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/archs/Padiff_arch/wavelet.py -------------------------------------------------------------------------------- /basicsr/archs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/archs/__init__.py -------------------------------------------------------------------------------- /basicsr/archs/arch_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/archs/arch_util.py -------------------------------------------------------------------------------- /basicsr/archs/wfdiffx2_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/archs/wfdiffx2_arch.py -------------------------------------------------------------------------------- /basicsr/data/MambaIR_paired_image_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/data/MambaIR_paired_image_dataset.py -------------------------------------------------------------------------------- /basicsr/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/data/__init__.py -------------------------------------------------------------------------------- /basicsr/data/data_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/data/data_sampler.py -------------------------------------------------------------------------------- /basicsr/data/data_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/data/data_util.py -------------------------------------------------------------------------------- /basicsr/data/paired_image_dataset copy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/data/paired_image_dataset copy.py -------------------------------------------------------------------------------- /basicsr/data/paired_image_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/data/paired_image_dataset.py -------------------------------------------------------------------------------- /basicsr/data/prefetch_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/data/prefetch_dataloader.py -------------------------------------------------------------------------------- /basicsr/data/single_image_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/data/single_image_dataset.py -------------------------------------------------------------------------------- /basicsr/data/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/data/transforms.py -------------------------------------------------------------------------------- /basicsr/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/losses/__init__.py -------------------------------------------------------------------------------- /basicsr/losses/loss_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/losses/loss_util.py -------------------------------------------------------------------------------- /basicsr/losses/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/losses/losses.py -------------------------------------------------------------------------------- /basicsr/metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/metrics/__init__.py -------------------------------------------------------------------------------- /basicsr/metrics/metric_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/metrics/metric_util.py -------------------------------------------------------------------------------- /basicsr/metrics/psnr_ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/metrics/psnr_ssim.py -------------------------------------------------------------------------------- /basicsr/models/Wfdiff_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/models/Wfdiff_model.py -------------------------------------------------------------------------------- /basicsr/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/models/__init__.py -------------------------------------------------------------------------------- /basicsr/models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/models/base_model.py -------------------------------------------------------------------------------- /basicsr/models/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/models/lr_scheduler.py -------------------------------------------------------------------------------- /basicsr/models/sr_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/models/sr_model.py -------------------------------------------------------------------------------- /basicsr/models/wavelet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/models/wavelet.py -------------------------------------------------------------------------------- /basicsr/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/test.py -------------------------------------------------------------------------------- /basicsr/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/train.py -------------------------------------------------------------------------------- /basicsr/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/utils/__init__.py -------------------------------------------------------------------------------- /basicsr/utils/diff_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/utils/diff_util.py -------------------------------------------------------------------------------- /basicsr/utils/dist_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/utils/dist_util.py -------------------------------------------------------------------------------- /basicsr/utils/file_client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/utils/file_client.py -------------------------------------------------------------------------------- /basicsr/utils/img_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/utils/img_util.py -------------------------------------------------------------------------------- /basicsr/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/utils/logger.py -------------------------------------------------------------------------------- /basicsr/utils/matlab_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/utils/matlab_functions.py -------------------------------------------------------------------------------- /basicsr/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/utils/misc.py -------------------------------------------------------------------------------- /basicsr/utils/options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/utils/options.py -------------------------------------------------------------------------------- /basicsr/utils/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/utils/registry.py -------------------------------------------------------------------------------- /basicsr/version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/basicsr/version.py -------------------------------------------------------------------------------- /options/test/test_wfdiff.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/options/test/test_wfdiff.yml -------------------------------------------------------------------------------- /options/train/train_Wfdiff.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/options/train/train_Wfdiff.yml -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChenzhaoNju/WF-Diff/HEAD/setup.py --------------------------------------------------------------------------------