Skip to content
This repository has been archived by the owner on Jan 11, 2023. It is now read-only.

edezhic/fashion-generator

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Fashion Generator

GAN model trained in Tensorflow and then ported to browsers using GPU-accelarated framework deeplearn.js which is now the core of tensorflow.js

You can repeat the whole experiment in 3 steps:

  • Run train.​py to get ./weights folder
  • Build generator.ts
  • Run node server and open index.html
  • You are awesome!
./model contents:
  • train.​py - creates and trains the model, saves generator weights
  • GAN.​py - model definition (originated from this repo)
  • yellowfin.​py - powerful custom TF optimizer, original repo. Not really required, just a nice thing.
  • ops.​py and utils.​py - simplified TF operations and other utility functions
./browser contents:
  • generator.ts - downloads weights and defines generator model
  • index.html - simple wrapper
  • other configs

About

In-browser GPU-accelerated Generative Adversarial Network trained on Fashion-MNIST dataset (tensorflow + deeplearn.js)

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published