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