← all repositories
taki0112/UGATIT

Unsupervised image translation that reshapes, not just recolors

U-GAT-IT uses attention maps and adaptive normalization to handle large geometric changes between domains without paired training data.

6.1k stars Python Image · Video · Audio
UGATIT
Not currently ranked — collecting fresh signals.
star history

What it does

U-GAT-IT is an unsupervised image-to-image translation system that learns to map between two visual domains—such as selfies and anime portraits—without paired examples. Unlike simpler style-transfer approaches, it attempts to handle translations that require actual geometric deformation, not just texture and color shifts.

The interesting bit

Most attention-based translation models fail when source and target domains have different shapes. U-GAT-IT introduces an attention module guided by an auxiliary classifier to focus on regions that actually distinguish the domains, paired with AdaLIN—Adaptive Layer-Instance Normalization—which learns per-dataset how much to deform shape versus texture. The paper reports superiority over fixed-architecture state-of-the-art models using a single set of hyper-parameters.

Key highlights

  • Handles large shape changes between domains, not just holistic style transfers
  • AdaLIN learns dataset-specific parameters to balance geometric deformation and texture preservation
  • Attention maps are derived from an auxiliary classifier, not hand-engineered rules
  • Pretrained selfie2anime checkpoints released for 50 and 100 epochs
  • Community has already built web demos and a Telegram bot on top of it

Caveats

  • Locked to TensorFlow 1.14 and Python 3.6, so modern environments will need careful dependency reconstruction
  • The reduced --light mode for limited GPU memory explicitly warns it may not perform well
  • Full training requires hardware generous enough to run the paper configuration

Verdict

Useful as a research baseline for geometric-aware unsupervised translation or for reproducing the ICLR 2020 benchmark. If you need a current, production-ready image translator, treat this as a research artifact rather than a modern stack.

Frequently asked

What is taki0112/UGATIT?
U-GAT-IT uses attention maps and adaptive normalization to handle large geometric changes between domains without paired training data.
Is UGATIT open source?
Yes — taki0112/UGATIT is open source, released under the MIT license.
What language is UGATIT written in?
taki0112/UGATIT is primarily written in Python.
How popular is UGATIT?
taki0112/UGATIT has 6.1k stars on GitHub.
Where can I find UGATIT?
taki0112/UGATIT is on GitHub at https://github.com/taki0112/UGATIT.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.