cswry / OSEDiff

75 stars 0 forks source link

One-Step Effective Diffusion Network for Real-World Image Super-Resolution

Rongyuan Wu1,2,*Lingchen Sun1,2,*Zhiyuan Ma1,*Lei Zhang1,2,†
1The Hong Kong Polytechnic University, 2OPPO Research Institute 
[[paper]](https://arxiv.org/pdf/2406.08177v2) ---

🔥 News

🎬 Overview

overview

🔧 Dependencies and Installation

  1. Clone repo

    git clone https://github.com/cswry/OSEDiff.git
    cd OSEDiff
  2. Install dependent packages

    conda create -n OSEDiff python=3.10 -y
    conda activate OSEDiff
    pip install --upgrade pip
    pip install -r requirements.txt
  3. Download Models

    Dependent Models

⚡ Quick Inference

python test_osediff.py \
-i preset/datasets/test_dataset/input \
-o preset/datasets/test_dataset/output \
--osediff_path preset\models\osediff.pkl \
--pretrained_model_name_or_path SD21BASE_PATH \
--ram_ft_path DAPE_PATH \
--ram_path RAM_PATH

📷 Results

benchmark

Quantitative Comparisons (click to expand)

Visual Comparisons (click to expand)

📧 Contact

If you have any questions, please feel free to contact: rong-yuan.wu@connect.polyu.hk

🎓Citations

@article{wu2024one,
  title={One-Step Effective Diffusion Network for Real-World Image Super-Resolution},
  author={Wu, Rongyuan and Sun, Lingchen and Ma, Zhiyuan and Zhang, Lei},
  journal={arXiv preprint arXiv:2406.08177},
  year={2024}
}
statistics ![visitors](https://visitor-badge.laobi.icu/badge?page_id=cswry/OSEDiff)