| | --- |
| | license: mit |
| | language: |
| | - en |
| | --- |
| | # 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](https://colab.research.google.com/github/louwjac/CycleGAN-Flax/blob/main/notebooks/inference.ipynb) 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](https://huggingface.co/spaces/louwjac/CycleGan-Flax) 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](https://tensorboard.dev/experiment/5tJZAfBLQkOJ66BUrBVBkQ/) |
| | 2. [monet2photo](https://tensorboard.dev/experiment/GZZKiDoBTl6dsgsqRo7rZw/) |
| |
|
| |
|
| | <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;'> |
| |
|
