TASR: Timestep-Aware Diffusion Model for Image Super-Resolution

Paper

Installation

# clone this repo
git clone https://github.com/SleepyLin/TASR.git
cd TASR
# create environment
conda create -n tasr python=3.10
conda activate tasr
pip install -r requirements.txt

Pretrained Models

todo

Inference

1. Download pretrained Stable Diffusion v2.1 and BSRNet
2. Download pretrained TASR v1 Model
3. Set correct model path in shell script

sh scripts/inference.sh

Train

todo Set ./config/train_tast.yaml

sh scripts/train.sh

Citation

Please cite us if our work is useful for your research.

@misc{lin2024tasrtimestepawarediffusionmodel,
      title={TASR: Timestep-Aware Diffusion Model for Image Super-Resolution}, 
      author={Qinwei Lin and Xiaopeng Sun and Yu Gao and Yujie Zhong and Dengjie Li and Zheng Zhao and Haoqian Wang},
      year={2024},
      eprint={2412.03355},
      archivePrefix={arXiv},
      primaryClass={cs.CV},
      url={https://arxiv.org/abs/2412.03355}, 
}

Thanks

This project is based on DiffBIR. Thanks for their awesome work.