CycleGAN-Flax

CycleGAN in Jax + Flax

This repository contains weights for CycleGAN models that were trained on the horse2zebra and monet2photo TensorFlow datasets in Jax with Flax. Please see the Colab notebook for an example of how to instantiate the models and load the trained weights. It includes a Gradio demo that makes it easy to submit external images to the model. You can also test the models in the Spaces demo without running any code yourself.

TensorBoard logs

The TensorBoard logs for the training runs can be viewed at the links below. The logged images are unfortunately not available due to file size constraints.

1. horse2zebra
2. monet2photo

<a href="https://huggingface.co/spaces/louwjac/CycleGan-Flax"> <img src="https://huggingface.co/datasets/huggingface/badges/raw/main/open-in-hf-spaces-sm-dark.svg" alt="Open in Spaces" style='display:inline;'> </a> <a target="_blank" href="https://colab.research.google.com/github/louwjac/CycleGAN-Flax/blob/main/notebooks/inference.ipynb"> <img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab" style='display:inline;'/> </a> <a href="https://github.com/louwjac/CycleGAN-Flax"> <img alt="Custom badge" src="https://img.shields.io/static/v1?style=flat&logo=github&logoColor=white&labelColor=black&label=&message=View%20on%20GitHub&color=white" style='display:inline;'>