/OneGan

One more freaking GAN, I'll be sick.

Primary LanguagePythonMIT LicenseMIT

OneGAN

https://travis-ci.com/leVirve/OneGAN.svg?token=6EpaKFxfJsjbzMkoLpeL&branch=master

Oh, GAN! One more freaking GAN i'll be sick. Here's yet another PyTorch framework for GAN and plain network.

Features

  • Out-of-the-box trainers/estimators, don't do it yourself
  • Performance measurement for semantic segmentation and image assessment
  • Extenstions for the storage of intermediate/result images and checkpoints
  • Visualization extension for TensorBoard
  • Event handlers designs ignited by ignite

Prerequsite

  • PyTorch >= 0.4
  • torchvision >= 0.2.1

Installation

python setup.py install