Skip to content

Instantly share code, notes, and snippets.

@NMZivkovic

NMZivkovic/usage_cycle_gan.py

Last active February 10, 2019 18:30
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save NMZivkovic/c8d5e0775b7fb26c096cb19489e40453 to your computer and use it in GitHub Desktop.
Save NMZivkovic/c8d5e0775b7fb26c096cb19489e40453 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
usage_cycle_gan.py
import numpy as np
from glob import glob
from image_helper_cycle_gan import ImageHelper
from cycle_gan import CycleGAN
image_helper = ImageHelper()
print("Ploting the images...")
filenames = np.array(glob('monet2photo/testA/*.jpg'))
image_helper.plot20(filenames)
generative_advarsial_network = CycleGAN((128, 128, 3), 10.0, image_helper)
generative_advarsial_network.train(100, 1, "monet2photo")
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment