LibreAI / Изображения / contrastive-unpaired-translation
// image · taesungp/contrastive-unpaired-translation

contrastive-unpaired-translation

contrastive-unpaired-translation — проект с открытым исходным кодом из категории «Изображения» от taesungp (taesungp/contrastive-unpaired-translation на GitHub). У него 2 556 звёзд на GitHub и 452 форка, он распространяется по лицензии NOASSERTION и последний раз обновлялся на GitHub 5 сентября 2023.

CUT — contrastive unpaired image-to-image translation in PyTorch, faster and lighter than CycleGAN (ECCV 2020)

★ 2,6 тыс. звёзд⑂ 452 форкаNOASSERTIONобновлено сент. 2023
▦ визуал — в README этого проекта нет встраиваемого скриншота,
поэтому здесь заглушка. Руководство ниже в любом случае полное.
// факты о проекте

Факты о проекте

Звёзды GitHub
2 556
Форки
452
Лицензия
NOASSERTION
Последнее обновление на GitHub
5 сентября 2023
Категория
Изображения
Тип страницы
Полное руководство по установке
Официальный сайт
Official site
// как это работает

Как это работает

  1. Clarify your goal

    Gather two unpaired image collections (domains A and B) — e.g. horses and zebras — or download one of the provided datasets; decide CUT vs FastCUT vs single-image SinCUT.

  2. Clone the repo

    Clone into a folder named CUT, as the README's getting-started shows.

  3. Install dependencies

    pip install -r requirements.txt (or conda env create -f environment.yml); optionally start a visdom server to watch training.

  4. Train and test

    Download a dataset with the provided script, train with train.py, then generate results with test.py — or skip training using the pretrained models and launcher scripts.

// возможности

Почему его любят

PatchNCE contrastive loss

Patchwise contrastive learning ties corresponding patches of input and output together, replacing cycle consistency and its inverse network.

Faster, lighter than CycleGAN

The authors report faster training and lower memory than CycleGAN, with stronger distribution matching in their experiments.

FastCUT variant

A lighter configuration (no identity loss, higher lambda_NCE) described as using half the GPU memory and training twice as fast as CycleGAN.

Single-image translation (SinCUT)

Each 'domain' can be just one image, using StyleGAN2-based components adopted in the repo.

Pretrained models

Downloadable pretrained checkpoints (Cityscapes, Horse2Zebra, Cat2Dog) with launcher scripts and FID evaluation instructions.

Launcher scripts

Experiment launchers generate the long train/test command lines for the provided datasets.

Advertisement
// запустите сами

Запустите сами

  1. Clone the repo

    Clone CUT into a directory named CUT, as the README shows.

    git clone https://github.com/taesungp/contrastive-unpaired-translation.git
    cd contrastive-unpaired-translation
  2. Install dependencies

    Install PyTorch and the other requirements with pip (or use the conda environment file).

    pip install -r requirements.txt
  3. Train and test

    Download the example dataset, train CUT, then run the test script.

    bash ./datasets/download_cut_dataset.sh grumpifycat
    python train.py --dataroot ./datasets/grumpifycat --name grumpycat_CUT --CUT_mode CUT
    python test.py --dataroot ./datasets/grumpifycat --name grumpycat_CUT --CUT_mode CUT --phase train
// сценарии использования

Что можно создать

CUT exists because of the work it streamlines: the unpaired translation problem was defined by CycleGAN, in the classic PyTorch codebase that also carries pix2pix — still the implementation most translation projects are measured against.

Advertisement
// вопросы и ответы

Вопросы и ответы

Who maintains CUT?

The authors — Taesung Park, Alexei A. Efros, Richard Zhang and Jun-Yan Zhu (UC Berkeley / Adobe Research) — published it with their ECCV 2020 paper.

Is it free?

The code is public research code; GitHub lists its license as NOASSERTION, so read the repository's license file before reusing it commercially.

Does it need a paid API key?

No — training and inference run locally in PyTorch on your own GPU.

How does it compare to CycleGAN?

CycleGAN enforces cycle consistency with a second generator. CUT uses a contrastive PatchNCE loss instead, which the authors show trains faster with less memory; FastCUT is the even lighter option.

How do I get started?

Clone, pip install -r requirements.txt, download the grumpifycat dataset, and run the train.py command from the README.

// продолжить изучение

Похожие инструменты

Изображения

pytorch-CycleGAN-and-pix2pix

The classic PyTorch implementation of CycleGAN and pix2pix for image-to-image translation

★ 25,3 тыс. звёздNOASSERTION
Смотреть руководство →
Изображения

diffusers

🤗 Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch.

★ 34,6 тыс. звёздApache-2.0
Смотреть руководство →
Изображения

DragGAN

Unofficial DragGAN implementation: drag points on a GAN image to edit it interactively

★ 4,9 тыс. звёздNOASSERTION
Смотреть руководство →
Изображения

EcomGen

Local-first AI e-commerce image suite generator for solo sellers

★ 133 звездыMIT
Смотреть руководство →
Изображения

pollinations-image-lab

Modern AI image lab on Pollinations.ai — models, seeds, history, transparency

★ 117 звёздMIT
Смотреть руководство →
Изображения

pixelshift-editor

Desktop image editor with AI effects, built for modern apps

★ 114 звёздCustom
Смотреть руководство →
// альтернативы

Альтернативы: Изображения

Другие проекты из категории «Изображения» с наибольшим числом звёзд в каталоге LibreAI — реалистичный список альтернатив, если contrastive-unpaired-translation вам не подходит.

// ещё в категории image

Больше проектов: Изображения

Соседние проекты из категории «Изображения» в каталоге, ближайшие по числу звёзд к contrastive-unpaired-translation — продолжайте изучать полку по обе стороны от него.

Получить contrastive-unpaired-translation

Бесплатно и открыто — забирайте исходный код, ставьте звезду репозиторию и начинайте создавать.