# ComfyUI-DiffBIR

**Repository Path**: dxal0001/ComfyUI-DiffBIR

## Basic Information

- **Project Name**: ComfyUI-DiffBIR
- **Description**: ComfyUI的一个节点DiffBIR，来源：【多人老照片修复ComfyUI免费教学工作流分享】
https://www.bilibili.com/video/BV1bsUSYSEar?t=122.7
- **Primary Language**: Python
- **License**: Not specified
- **Default Branch**: main
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2025-01-07
- **Last Updated**: 2025-01-07

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# ComfyUI DiffBIR

Comfyui-DiffBIR is a comfyui implementation of offical DiffBIR. DiffBIR v2 is an awesome super-resolution algorithm. If you find this repo helpful, please don't hesitate to give it a star. Thank you! ❤️❤️❤️

---

TODO:
- **2024.05.10** ✅ repo create
- **2024.05.11** ✅ sampler implement
- **2024.05.11** ✅ stage2 load implement
- **2024.05.12** ✅ stage1 load implement
- **2024.05.12** ✅ stage1 tile implement
- **2024.05.12** ✅ simple load implement
- **2024.05.12** ✅ sampler update
- **2024.05.12** ✅ sampler advanced
- **2024.05.14** ✅ readme update
- **2024.05.14** ✅ bsr workflow
- **2024.05.16** ✅ bfr support
- **2024.05.16** ✅ bid support
- **2024.05.16** ✅ bfr and bid workflow
- **2024.05.21** ✅ support fp16 inference, now it can run on an 8GB GPU
- [ ] multi images support
- [ ] stage1 and stage2 sampler split

# Visual Results
## Blind Image Super-Resolution
|        low resolution image       |       super resolution image      |
|:---------------------------------:|:---------------------------------:|
|   ![2_lq](./assets/bsr/2_lq.jpg ) | ![ 2_hq ]( ./assets/bsr/2_hq.png )|
|   ![1_lq](./assets/bsr/1_lq.jpg)  | ![ 1_hq ]( ./assets/bsr/1_hq.png )|

## Blind Face Restoration
|        low resolution image       |       super resolution image      |
|:---------------------------------:|:---------------------------------:|
|   ![1_lq](./assets/bfr/1_lq.png ) | ![ 1_hq ]( ./assets/bfr/1_hq.png )|
|   ![2_lq](./assets/bfr/2_lq.jpg)  | ![ 2_hq ]( ./assets/bfr/2_hq.png )|

## Blind Image Denoising
|        low resolution image       |       super resolution image      |
|:---------------------------------:|:---------------------------------:|
|   ![1_lq](./assets/bid/1_lq.jpg ) | ![ 1_hq ]( ./assets/bid/1_hq.png )|
|   ![2_lq](./assets/bid/2_lq.png)  | ![ 2_hq ]( ./assets/bid/2_hq.png )|

# Installing
### Step 1: clone the repo and install the dependencies
To install ComfyUI-DiffBIR in addition to an existing installation of ComfyUI, you can follow the following steps:
1. `cd custom_nodes`  
2. `git clone https://github.com/ComfyUI/ComfyUI-DiffBIR`  
3. `pip install -r requirements.txt`
4. Restart ComfyUI

### Step 2: download the pretrained model
put the models into `ComfyUI/models/diffbir/`
|      model     |                                                          download link                                                         | Baidu Netdisk |
|:--------------:|:------------------------------------------------------------------------------------------------------------------------------:|---------------|
|     bsrnet     | [BSRNet.pth](https://github.com/cszn/KAIR/releases/download/v1.0/BSRNet.pth)                                                   |               |
|   swinir_face  | [face_swinir_v1.ckpt](https://huggingface.co/lxq007/DiffBIR/resolve/main/face_swinir_v1.ckpt)                                  |               |
|   scunet_psnr  |          [scunet_color_real_psnr.pth](https://github.com/cszn/KAIR/releases/download/v1.0/scunet_color_real_psnr.pth)          |               |
| swinir_general | [general_swinir_v1.ckpt](https://huggingface.co/lxq007/DiffBIR/resolve/main/general_swinir_v1.ckpt)                            |               |
|       v2       | [v2.pth](https://huggingface.co/lxq007/DiffBIR-v2/resolve/main/v2.pth)                                                         |               |
|     sd_v21     | [v2-1_512-ema-pruned.ckpt](https://huggingface.co/stabilityai/stable-diffusion-2-1-base/resolve/main/v2-1_512-ema-pruned.ckpt) |               |


# Workflows
|           task           |                              workflow                                  |
|:------------------------:|:----------------------------------------------------------------------:|
| Blind Super Resolution   |  [  bsr_workflow.json  ](  example_workflows/bsr_workflow_fp32.json  ) |
| Blind Face Restoration   |  [  bfr_workflow.json  ](  example_workflows/bfr_workflow_fp32.json  ) |
| Blind Image Denoising    |  [  bid_workflow.json  ](  example_workflows/bid_workflow_fp32.json  ) |