BSRD (Burst Super-Resolution with Diffsuion Model)

This is the official repository for the following paper:

Kyotaro Tokoro, Kazutoshi Akita, and Norimichi Ukita

Burst Super-Resolution with Diffusion Models for Improving Perceptual Quality, IJCNN2024

SyntheticBurst Dataset

sh train_synth.sh

Download syn_burst_val and place it in './synth/SyntheticBurstVal'.
Download BIPNet Trained model or BIPNet and place it in './Trained_models/synth/BIPNet.ckpt'.
Download Burstormer Trained model or Burstormer and place it in './Trained_models/synth/Burstormer.ckpt'.
Download BSRD Trained model and place it in './Trained_models/synth/BSRD_synth.pt'.

sh sample_synth.sh

sh train_real.sh

Download burstsr_dataset.
Download PWC Trained model and place it in './pwcnet/pwcnet-network-default.pth'.
Download BIPNet Trained model or BIPNet and place it in './Trained_models/real/BIPNet.ckpt'.
Download Burstormer Trained model or Burstormer and place it in './Trained_models/real/Burstormer.ckpt'.
Download BSRD Trained model and place it in './Trained_models/real/BSRD_real.pt'.

sh sample_real.sh

Name		Name	Last commit message	Last commit date
Latest commit History 6 Commits
bip		bip
burstormer		burstormer
data_processing		data_processing
datasets		datasets
evaluations		evaluations
guided_diffusion		guided_diffusion
image		image
pwcnet		pwcnet
utils		utils
.gitignore		.gitignore
LICENSE		LICENSE
README.md		README.md
requirements.txt		requirements.txt
sample_real.py		sample_real.py
sample_real.sh		sample_real.sh
sample_synth.py		sample_synth.py
sample_synth.sh		sample_synth.sh
setup.py		setup.py
train_diffusion_real.py		train_diffusion_real.py
train_diffusion_synth.py		train_diffusion_synth.py
train_real.sh		train_real.sh
train_synth.sh		train_synth.sh