metadata
license: mit
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
from transformers import AutoModel
model = AutoModel.from_pretrained("vinhsmoke/image_reflection_removal", trust_remote_code = True)