├── Datasets └── README.md ├── LICENSE ├── README.md ├── Weights └── README.md ├── assets ├── MPT.png └── catablur.png ├── ckpt_manager.py ├── configs ├── config.py └── config_MPT.py ├── data_loader ├── DP_datasets_lr_aug.py ├── FastDataLoader.py ├── data_sampler.py └── utils.py ├── eval.py ├── models ├── MPT.py ├── __init__.py ├── archs │ └── LPIPS │ │ ├── __init__.py │ │ ├── base_model.py │ │ ├── dist_model.py │ │ ├── networks_basic.py │ │ ├── pretrained_networks.py │ │ └── weights │ │ └── v0.1 │ │ ├── alex.pth │ │ ├── squeeze.pth │ │ └── vgg.pth ├── baseModel.py ├── loss.py ├── nn_common.py ├── ops.py ├── trainers │ ├── lion.py │ ├── lr_scheduler.py │ ├── metrics.py │ ├── psnr_ssim.py │ └── trainer.py └── utils.py ├── requirements.txt ├── run.py ├── run_inference.py ├── train_dist.sh └── utils.py /Datasets/README.md: -------------------------------------------------------------------------------- 1 | Put all downloaded dataset under this folder. -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PieceZhang/MPT-CataBlur/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PieceZhang/MPT-CataBlur/HEAD/README.md -------------------------------------------------------------------------------- /Weights/README.md: -------------------------------------------------------------------------------- 1 | Put all downloaded weight files under this folder. -------------------------------------------------------------------------------- /assets/MPT.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PieceZhang/MPT-CataBlur/HEAD/assets/MPT.png -------------------------------------------------------------------------------- /assets/catablur.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PieceZhang/MPT-CataBlur/HEAD/assets/catablur.png -------------------------------------------------------------------------------- /ckpt_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PieceZhang/MPT-CataBlur/HEAD/ckpt_manager.py -------------------------------------------------------------------------------- /configs/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PieceZhang/MPT-CataBlur/HEAD/configs/config.py -------------------------------------------------------------------------------- /configs/config_MPT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PieceZhang/MPT-CataBlur/HEAD/configs/config_MPT.py -------------------------------------------------------------------------------- /data_loader/DP_datasets_lr_aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PieceZhang/MPT-CataBlur/HEAD/data_loader/DP_datasets_lr_aug.py -------------------------------------------------------------------------------- /data_loader/FastDataLoader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PieceZhang/MPT-CataBlur/HEAD/data_loader/FastDataLoader.py -------------------------------------------------------------------------------- /data_loader/data_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PieceZhang/MPT-CataBlur/HEAD/data_loader/data_sampler.py -------------------------------------------------------------------------------- /data_loader/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PieceZhang/MPT-CataBlur/HEAD/data_loader/utils.py -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PieceZhang/MPT-CataBlur/HEAD/eval.py -------------------------------------------------------------------------------- /models/MPT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PieceZhang/MPT-CataBlur/HEAD/models/MPT.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PieceZhang/MPT-CataBlur/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/archs/LPIPS/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PieceZhang/MPT-CataBlur/HEAD/models/archs/LPIPS/__init__.py -------------------------------------------------------------------------------- /models/archs/LPIPS/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PieceZhang/MPT-CataBlur/HEAD/models/archs/LPIPS/base_model.py -------------------------------------------------------------------------------- /models/archs/LPIPS/dist_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PieceZhang/MPT-CataBlur/HEAD/models/archs/LPIPS/dist_model.py -------------------------------------------------------------------------------- /models/archs/LPIPS/networks_basic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PieceZhang/MPT-CataBlur/HEAD/models/archs/LPIPS/networks_basic.py -------------------------------------------------------------------------------- /models/archs/LPIPS/pretrained_networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PieceZhang/MPT-CataBlur/HEAD/models/archs/LPIPS/pretrained_networks.py -------------------------------------------------------------------------------- /models/archs/LPIPS/weights/v0.1/alex.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PieceZhang/MPT-CataBlur/HEAD/models/archs/LPIPS/weights/v0.1/alex.pth -------------------------------------------------------------------------------- /models/archs/LPIPS/weights/v0.1/squeeze.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PieceZhang/MPT-CataBlur/HEAD/models/archs/LPIPS/weights/v0.1/squeeze.pth -------------------------------------------------------------------------------- /models/archs/LPIPS/weights/v0.1/vgg.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PieceZhang/MPT-CataBlur/HEAD/models/archs/LPIPS/weights/v0.1/vgg.pth -------------------------------------------------------------------------------- /models/baseModel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PieceZhang/MPT-CataBlur/HEAD/models/baseModel.py -------------------------------------------------------------------------------- /models/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PieceZhang/MPT-CataBlur/HEAD/models/loss.py -------------------------------------------------------------------------------- /models/nn_common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PieceZhang/MPT-CataBlur/HEAD/models/nn_common.py -------------------------------------------------------------------------------- /models/ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PieceZhang/MPT-CataBlur/HEAD/models/ops.py -------------------------------------------------------------------------------- /models/trainers/lion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PieceZhang/MPT-CataBlur/HEAD/models/trainers/lion.py -------------------------------------------------------------------------------- /models/trainers/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PieceZhang/MPT-CataBlur/HEAD/models/trainers/lr_scheduler.py -------------------------------------------------------------------------------- /models/trainers/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PieceZhang/MPT-CataBlur/HEAD/models/trainers/metrics.py -------------------------------------------------------------------------------- /models/trainers/psnr_ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PieceZhang/MPT-CataBlur/HEAD/models/trainers/psnr_ssim.py -------------------------------------------------------------------------------- /models/trainers/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PieceZhang/MPT-CataBlur/HEAD/models/trainers/trainer.py -------------------------------------------------------------------------------- /models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PieceZhang/MPT-CataBlur/HEAD/models/utils.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PieceZhang/MPT-CataBlur/HEAD/requirements.txt -------------------------------------------------------------------------------- /run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PieceZhang/MPT-CataBlur/HEAD/run.py -------------------------------------------------------------------------------- /run_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PieceZhang/MPT-CataBlur/HEAD/run_inference.py -------------------------------------------------------------------------------- /train_dist.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PieceZhang/MPT-CataBlur/HEAD/train_dist.sh -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PieceZhang/MPT-CataBlur/HEAD/utils.py --------------------------------------------------------------------------------