xiaoojs20's picture
Update README.md
a113f34 verified
metadata
license: mit

CUT Monet Style Transfer (Photo → Monet)

This repository contains the Best-Restart CUT model used in the
Kaggle competition "I'm Something of a Painter Myself".

🎉 Our team achieved
36.94468 score, ranking 4th / 170 teams (Top 2%).

Model

We provide the trained CUT Generator A (Photo→Monet) checkpoint.

  • Architecture: CUT (Contrastive Unpaired Translation)
  • Trained resolution: 256×256
  • Restart strategy: Best-Restart (achieved best FID ≈ 1.6)

Usage

from inference import load_model, inference

model = load_model("generator_A.pth")
inference(model, "photo.jpg", "monet_output.png")