# ⚛️ GAN-MASTER

An optimized AI tool support image restoration, image denoise, upscale, ... using GFP-GAN, GPEN, ... 

> **⚠️ Important Note**  
> 🌵 I only have RTX 30 series (SM86). So newer series untested.  
> 🌵 If you facing any issue, please send the error log to:  
>      💮 [LinkedIn](https://www.linkedin.com/in/huu-san-dich-91a079119/)   
>      💮 [Facebook](https://www.facebook.com/shuugame/)  
>      💮 [Whatsapp](https://wa.me/84328071096)  

<br>

<table style="border: none; border-collapse: collapse;">
  <tr style="border: none;">
    <td style="border: none; padding: 5px 10px;"><strong>🌵 Compatible GPU</strong></td>
    <td style="border: none; padding: 5px 10px;"></td>
  </tr>
  <tr style="border: none;">
    <td style="border: none; padding: 5px 10px;">✅ <strong>Newer</strong></td>
    <td style="border: none; padding: 5px 10px;">Runnable, but slower.</td>
  </tr>
  <tr style="border: none;">
    <td style="border: none; padding: 5px 10px;">✅ <strong>SM86</strong></td>
    <td style="border: none; padding: 5px 10px;">RTX 30 Series.</td>
  </tr>
  <tr style="border: none;">
    <td style="border: none; padding: 5px 10px;">⛔ <strong>Unsupported</strong></td>
    <td style="border: none; padding: 5px 10px;">RTX 20 Series and older, AMD or Intel GPU.</td>
  </tr>
</table>


## 🌸 Quick Start (Docker Compose)

Install **Docker**, **Newest Nvidia Driver & CUDA toolkit**   
Next, copy **docker-compose.yml** content below and put to empty folder.  
Then, open **Terminal**/**CMD**/**Windows PowerShell** and navigate to directory contain `docker-compose.yml` file.  
Finally, run this command `docker compose up -d` to start the container.
Go to `http://localhost:7860` and enjoy.

>### **Requirements**  
> ⬇️ [Docker](https://docs.docker.com/desktop)  
> ⬇️ [nvidia-driver](https://www.nvidia.com/en-us/drivers)  
> ⬇️ [cuda-toolkit](https://developer.nvidia.com/cuda-downloads)  
> 
> ### **Models (download then copy into models folder**
> ⬇️ [RetinaFace-R50](https://public-vigen-video.oss-cn-shanghai.aliyuncs.com/robin/models/RetinaFace-R50.pth)  
> ⬇️ [ParseNet-latest](https://public-vigen-video.oss-cn-shanghai.aliyuncs.com/robin/models/ParseNet-latest.pth)  
> ⬇️ [realesrnet_x4](https://public-vigen-video.oss-cn-shanghai.aliyuncs.com/robin/models/realesrnet_x4.pth)  
> ⬇️ [GPEN-BFR-512](https://public-vigen-video.oss-cn-shanghai.aliyuncs.com/robin/models/GPEN-BFR-512.pth)  
> ⬇️ [Colorization](https://public-vigen-video.oss-cn-shanghai.aliyuncs.com/robin/models/GPEN-Colorization-1024.pth)  
> ⬇️ [RealESRGAN_x2plus](https://github.com/xinntao/Real-ESRGAN/releases/download/v0.2.1/RealESRGAN_x2plus.pth)  
> ⬇️ [GFPGANv1.4](https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.4.pth)  

### **docker-compose.yml**

```yaml
services:
  gan-master:
    image: sandichhuu/gan-master:sm86cu130
    container_name: gan-master
    ipc: host
    port:
      - 7860:7860
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]
    volumes:
      - ./models:/workspace/app/models
      - ./outputs:/workspace/app/outputs

```

## 🌸 Setup Structure

```
comfyuv/
├── docker-compose.yml
├── models/
└── output/
```

## 🌸 Screenshot

**GPEN (Restore Mode)**

[![](https://wiki.chenjia.org/uploads/images/gallery/2026-05/scaled-1680-/ak0RK9VksgxQPnKM-image-1779904970348.png)](https://wiki.chenjia.org/uploads/images/gallery/2026-05/ak0RK9VksgxQPnKM-image-1779904970348.png)

**GFPGAN (Upscale Mode)**

[![](https://wiki.chenjia.org/uploads/images/gallery/2026-05/scaled-1680-/fFaonFDtp3ntFgAh-image-1779905200731.png)](https://wiki.chenjia.org/uploads/images/gallery/2026-05/fFaonFDtp3ntFgAh-image-1779905200731.png)
---