https://github.com/znxlwm/funit-pytorch

Pytorch implementation of "Few-Shot Unsupervised Image-to-Image Translation" (ICCV 2019)
https://github.com/znxlwm/funit-pytorch

Last synced: 3 months ago
JSON representation

Pytorch implementation of "Few-Shot Unsupervised Image-to-Image Translation" (ICCV 2019)

• Host: GitHub
• URL: https://github.com/znxlwm/funit-pytorch
• Owner: znxlwm
• License: mit
• Created: 2019-08-14T01:01:31.000Z (about 6 years ago)
• Default Branch: master
• Last Pushed: 2019-08-14T02:09:51.000Z (about 6 years ago)
• Last Synced: 2024-09-05T15:48:29.588Z (about 1 year ago)
• Language: Python
• Size: 9.57 MB
• Stars: 36
• Watchers: 4
• Forks: 2
• Open Issues: 1
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

          
# FUNIT-pytorch

## : Few-Shot Unsupervised Image-to-Image Translation (ICCV 2019)



 

 



### [Paper](https://arxiv.org/abs/1905.01723) | [Official PyTorch code](https://github.com/NVlabs/FUNIT)

## Tensorflow Implementation

[Tensorflow Implementation](https://github.com/taki0112/FUNIT-Tensorflow)

## Usage

```

├── dataset

   └── YOUR_DATASET_NAME

       ├── train

           ├── class1 (class folder)

               ├── xxx.jpg (class1 image)

               ├── yyy.png

               ├── ...

           ├── class2

               ├── aaa.jpg (class2 image)

               ├── bbb.png

               ├── ...

           ├── class3

           ├── ...

       ├── test

           ├── class1

               ├── zzz.jpg (class1 image)

               ├── www.png

               ├── ...

           ├── class2

               ├── ccc.jpg (class2 image)

               ├── ddd.jpg

               ├── ...

```

### Train

```

> python main.py --name FUNIT --dataset YOUR_DATASET_NAME --phase train --K 1

```

### Test

```

> python main.py --name FUNIT --dataset YOUR_DATASET_NAME --phase test --K 1

```

## Architecture

![architecture](./assets/architecture.png)

## Our result

![our_result](./assets/our_result.jpg)

## Paper result

![paper_result](./assets/funit_example.jpg)