增强现实AR>> GANcedonia>> 返回
项目作者： burklight

项目描述 ：
Implementation of a cycle consistency GAN for object transfiguration applied to fruit images.
高级语言： Python
项目地址: git://github.com/burklight/GANcedonia.git
创建时间： 2018-03-23T12:54:28Z
项目社区：https://github.com/burklight/GANcedonia
开源协议：
下载


GANcedonia

In this repository you will find a cycle consistency GAN prepared to perform object transfiguration between different fruits.
This PyTorch implementation is based on the CycleGAN original paper from Jun-Yan Zhu, Taesung Park et al.
The main differences with the original paper are:

  • the usage of a different generator network (we use a residual network based on the good results from Leidig et al.). The latest implementation of the original authors also considers residual networks and U-nets.
  • the usage of odd weighted filters for the PatchGAN.

alt text

Authors

Dataset

The dataset used for this repository is the Kaggle Fruits 360 presented by Muresan and Oltean.
The images from this dataset are placed randomly over the textures in the textures folder with random scaling.

Prerequisites

  • PyTorch 0.3.0 or superior.
  • Python2 or Python3
  • If using an NVIDIA GPU: CUDA

Installation (Ubuntu)

  • Installation of CUDA
  1. wget developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-repo-ubuntu1604_9.1.85-1_amd64.deb
  2. sudo dpkg -i cuda-repo-ubuntu1604_9.1.85-1_amd64.deb
  3. sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub
  4. sudo apt-get update
  5. sudo apt-get install cuda
  • Installation of Pytorch 0.4 for Python 3.6
  1. sudo -H pip3 install http://download.pytorch.org/whl/cu91/torch-0.4.0-cp36-cp36m-linux_x86_64.whl 
  2. sudo -H pip3 install torchvision
  • Installation of Pytorch 0.4 for Python 3.5
  1. sudo -H pip3 install http://download.pytorch.org/whl/cu91/torch-0.4.0-cp36-cp35m-linux_x86_64.whl 
  2. sudo -H pip3 install torchvision