ID: 5d89f362a0171346e315e53e

Horse to Zebra GAN

by Deepak Mangla

Automatically converts horses to Zebras in images with the generative AI.


Contains Dockerfile Ā This model contains Dockerfile.


License: BSD 2

Tags: CycleGAN Horse to zebra

Ā Model stats and performance
Dataset Used View
Framework PyTorch
OS Used Linux
Publication View
Inference time in seconds per sample.

Screenshots


What is it?

This model automatically converts horses to Zebras in images with the generative AI.

How to use?

Using pip Prerequisites - cuda 10.1 and python 3.6

  • Install all requirements with command pip install -r requirements.txt.
  • Put one or more images of horses in 'Input' folder. (Initially sample images are provided for testing purpose).
  • Run python src/run.py -- input Input --output Output.

Using docker Prerequisites - Docker > 19.03

  • Build docker image and run docker container - docker build -t hor2zeb -f Dockerfile . && docker run -it hor2zeb
  • Run python src/run.py -- input Input --output Output .

Voila ! Horses have been converted to Zebras and images have been stored in 'Output' directory.

Stats:

Both cpu and gpu stats are provided in 'stats' folder.

Tested on:

  • python 3.5
  • ubuntu 16.04
  • cuda 10.1

Author View Profile


Deepak Mangla
Faridabad, India
Level 25 34010 XP

1304 Profile
Views

AI wizard who thinks with first principles

I am Deep Learning (Computer Vision) Engineer with specialization in enabling AI on the resource-constrained edge devices. I work on quantizations, pruning and efficient designing and deployment of DL networks. Read my article about Winograd - https://blog.usejournal.com/understanding-winograd-fast-convolution-a75458744ff

User Reviews



0 total ratings

Model has not been reviewed yet.

More by this user | Show All



Also checkout...