Skip to content

ravankhidirov/CycleGAN_winter_to_fall

Repository files navigation

CycleGAN_winter_to_fall

A model that converts fall images to winter and vice versa.

What does this model?

This model converts winter images to fall and vice versa. The training has been done for only 20 epochs, which took about 4 hours in google colab with GPU. The model is cycle consistent generative adversial neural network. (Originally this model has been used to convert horse images to zebra images) Here you can find some of the pictures that has been converted using this model: image

image

Where to find data?

Dataset is about 900 winter and 900 fall images collected by scraping of websites like Pixabay. In this task, Beautifull soup is used to scrape images. There are two zip files which are train and test images of winter and fall.

The links to download datasets:

train dataset: https://drive.google.com/file/d/1fKp01bKWR0ozXrBVmJ_APU03dZEXwwAJ/view?usp=sharing

test dataset: https://drive.google.com/file/d/1Wk7tX1jTYLqtTLeTCBVldJZc8Y7JJb_t/view?usp=sharing

How to use this model.

We have provided the dataset collected by web scraping. Addidtionally, we have saved our updated generators and discriminators. These can be downloaded and used as in LoadingSavedWeightsAndBiases.ipynb file. Datasets are in zip format. You can download and add them to your drive. After that the code will work in google colab.

Contributors

I would like to give special thanks to Tural (https://github.com/tu0ral). We worked together while building the model, training, and finding dataset.

About

A model that converts fall images to winter and vice versa.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published