CoolFace
Modelpublic

vinhsmoke/image_reflection_removal

sourceHugging Facemitupdated 1y agoView on Hugging Face
0likes8downloads
Model Card

Image Reflection Removal Model

Model Description

This is a Pix2Pix GAN model trained to remove reflections from images (e.g. windows, glasses). The generator is a U-Net with skip connections, trained on 512 image pairs.

Average Performance

  • —PSNR : 26.9058
  • —SSIM : 0.8881

Usage

python
from transformers import AutoModel
model = AutoModel.from_pretrained("vinhsmoke/image_reflection_removal", trust_remote_code = True)