├── Edinburgh.jpg ├── FBSDiff ├── dct_util.py ├── ddim_sampler.py ├── fbs_sampler.py ├── logger.py ├── model.py └── tools.py ├── LICENSE ├── README.md ├── demo.ipynb ├── imgs ├── method_overview.jpg ├── results.jpg └── teaser.jpg ├── inference.py ├── ldm ├── data │ ├── __init__.py │ └── util.py ├── models │ ├── autoencoder.py │ └── diffusion │ │ ├── __init__.py │ │ ├── ddim.py │ │ ├── ddpm.py │ │ ├── dpm_solver │ │ ├── __init__.py │ │ ├── dpm_solver.py │ │ └── sampler.py │ │ ├── plms.py │ │ └── sampling_util.py ├── modules │ ├── attention.py │ ├── diffusionmodules │ │ ├── __init__.py │ │ ├── model.py │ │ ├── openaimodel.py │ │ ├── upscaling.py │ │ └── util.py │ ├── distributions │ │ ├── __init__.py │ │ └── distributions.py │ ├── ema.py │ ├── encoders │ │ ├── __init__.py │ │ └── modules.py │ ├── image_degradation │ │ ├── __init__.py │ │ ├── bsrgan.py │ │ ├── bsrgan_light.py │ │ ├── utils │ │ │ └── test.png │ │ └── utils_image.py │ └── midas │ │ ├── __init__.py │ │ ├── api.py │ │ ├── midas │ │ ├── __init__.py │ │ ├── base_model.py │ │ ├── blocks.py │ │ ├── dpt_depth.py │ │ ├── midas_net.py │ │ ├── midas_net_custom.py │ │ ├── transforms.py │ │ └── vit.py │ │ └── utils.py └── util.py ├── models ├── README.md └── model_ldm_v15.yaml ├── openai └── README.md ├── painting.jpg ├── photo.jpg └── requirements.txt /Edinburgh.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/Edinburgh.jpg -------------------------------------------------------------------------------- /FBSDiff/dct_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/FBSDiff/dct_util.py -------------------------------------------------------------------------------- /FBSDiff/ddim_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/FBSDiff/ddim_sampler.py -------------------------------------------------------------------------------- /FBSDiff/fbs_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/FBSDiff/fbs_sampler.py -------------------------------------------------------------------------------- /FBSDiff/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/FBSDiff/logger.py -------------------------------------------------------------------------------- /FBSDiff/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/FBSDiff/model.py -------------------------------------------------------------------------------- /FBSDiff/tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/FBSDiff/tools.py -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/README.md -------------------------------------------------------------------------------- /demo.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/demo.ipynb -------------------------------------------------------------------------------- /imgs/method_overview.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/imgs/method_overview.jpg -------------------------------------------------------------------------------- /imgs/results.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/imgs/results.jpg -------------------------------------------------------------------------------- /imgs/teaser.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/imgs/teaser.jpg -------------------------------------------------------------------------------- /inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/inference.py -------------------------------------------------------------------------------- /ldm/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/data/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/ldm/data/util.py -------------------------------------------------------------------------------- /ldm/models/autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/ldm/models/autoencoder.py -------------------------------------------------------------------------------- /ldm/models/diffusion/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/models/diffusion/ddim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/ldm/models/diffusion/ddim.py -------------------------------------------------------------------------------- /ldm/models/diffusion/ddpm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/ldm/models/diffusion/ddpm.py -------------------------------------------------------------------------------- /ldm/models/diffusion/dpm_solver/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/ldm/models/diffusion/dpm_solver/__init__.py -------------------------------------------------------------------------------- /ldm/models/diffusion/dpm_solver/dpm_solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/ldm/models/diffusion/dpm_solver/dpm_solver.py -------------------------------------------------------------------------------- /ldm/models/diffusion/dpm_solver/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/ldm/models/diffusion/dpm_solver/sampler.py -------------------------------------------------------------------------------- /ldm/models/diffusion/plms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/ldm/models/diffusion/plms.py -------------------------------------------------------------------------------- /ldm/models/diffusion/sampling_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/ldm/models/diffusion/sampling_util.py -------------------------------------------------------------------------------- /ldm/modules/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/ldm/modules/attention.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/ldm/modules/diffusionmodules/model.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/openaimodel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/ldm/modules/diffusionmodules/openaimodel.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/upscaling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/ldm/modules/diffusionmodules/upscaling.py -------------------------------------------------------------------------------- /ldm/modules/diffusionmodules/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/ldm/modules/diffusionmodules/util.py -------------------------------------------------------------------------------- /ldm/modules/distributions/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/distributions/distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/ldm/modules/distributions/distributions.py -------------------------------------------------------------------------------- /ldm/modules/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/ldm/modules/ema.py -------------------------------------------------------------------------------- /ldm/modules/encoders/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/encoders/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/ldm/modules/encoders/modules.py -------------------------------------------------------------------------------- /ldm/modules/image_degradation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/ldm/modules/image_degradation/__init__.py -------------------------------------------------------------------------------- /ldm/modules/image_degradation/bsrgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/ldm/modules/image_degradation/bsrgan.py -------------------------------------------------------------------------------- /ldm/modules/image_degradation/bsrgan_light.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/ldm/modules/image_degradation/bsrgan_light.py -------------------------------------------------------------------------------- /ldm/modules/image_degradation/utils/test.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/ldm/modules/image_degradation/utils/test.png -------------------------------------------------------------------------------- /ldm/modules/image_degradation/utils_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/ldm/modules/image_degradation/utils_image.py -------------------------------------------------------------------------------- /ldm/modules/midas/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/midas/api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/ldm/modules/midas/api.py -------------------------------------------------------------------------------- /ldm/modules/midas/midas/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ldm/modules/midas/midas/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/ldm/modules/midas/midas/base_model.py -------------------------------------------------------------------------------- /ldm/modules/midas/midas/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/ldm/modules/midas/midas/blocks.py -------------------------------------------------------------------------------- /ldm/modules/midas/midas/dpt_depth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/ldm/modules/midas/midas/dpt_depth.py -------------------------------------------------------------------------------- /ldm/modules/midas/midas/midas_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/ldm/modules/midas/midas/midas_net.py -------------------------------------------------------------------------------- /ldm/modules/midas/midas/midas_net_custom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/ldm/modules/midas/midas/midas_net_custom.py -------------------------------------------------------------------------------- /ldm/modules/midas/midas/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/ldm/modules/midas/midas/transforms.py -------------------------------------------------------------------------------- /ldm/modules/midas/midas/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/ldm/modules/midas/midas/vit.py -------------------------------------------------------------------------------- /ldm/modules/midas/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/ldm/modules/midas/utils.py -------------------------------------------------------------------------------- /ldm/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/ldm/util.py -------------------------------------------------------------------------------- /models/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/models/README.md -------------------------------------------------------------------------------- /models/model_ldm_v15.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/models/model_ldm_v15.yaml -------------------------------------------------------------------------------- /openai/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/openai/README.md -------------------------------------------------------------------------------- /painting.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/painting.jpg -------------------------------------------------------------------------------- /photo.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/photo.jpg -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiangGao1102/FBSDiff/HEAD/requirements.txt --------------------------------------------------------------------------------