Giter Club home page Giter Club logo

stylegan_reimplementation's Introduction

This is a reimplementation of NVidia's stylegan https://github.com/NVlabs/stylegan I did for learning purposes. My priority was testing out changes such as non-square images and conditioning on labels (using acgan and projection discriminator) rather than keeping the code clean so right now it is a bit messy.

This has been tested, and supports features like rectangular images, but I am currently waiting on the most recent training session to finish before uploading results. Training and sample generation are both performed using 'run.py'.

It is mostly original, but includes a couple functions from the official implementation for comparison testing.

To run, see the comments in start_training.sh (and then run start_training.sh). This has only been tested in one environment, so feel free to create a github issue if you encounter a problem. This model can handle non-square images and I plan on organizing and including the tools I've made/modified to generate datasets like that soonish.

Blogs related to a tool I made to interact with StyleGAN models:

https://towardsdatascience.com/animating-ganime-with-stylegan-part-1-4cf764578e https://towardsdatascience.com/animating-ganime-with-stylegan-the-tool-c5a2c31379d

Original Paper: Karras, T., Laine, S., and Aila, T. A style-based generator architecture for generative adversarial networks

stylegan_reimplementation's People

Contributors

nolan-dev avatar

Stargazers

 avatar wangcl avatar  avatar Junying Huang avatar its me avatar  avatar SUNmode0586 avatar TranceDream avatar 老董 avatar zyf avatar  avatar mengliYuan avatar 西风逍遥游 avatar mathbbN avatar  avatar  avatar Minseong Kim avatar John Haugeland avatar Cameron Alexander avatar  avatar  avatar NOKUBI Takatsugu avatar 蔡舒起 avatar  avatar  avatar  avatar SNEHAL GHARAT avatar  avatar Alina Christenbury avatar Manjeet Singh avatar Steve Zhu avatar  avatar Riceball LEE avatar  avatar Brunvelop avatar  avatar Joohun, Maeng avatar Had avatar Fei avatar  avatar WPC avatar Ruipeng Zhang avatar  avatar Robert A. Gonsalves avatar  avatar John Sigmon avatar  avatar Cortez Ethridge avatar ipruning avatar  avatar Justin Huang avatar Evan Chong avatar  avatar  avatar Andrew Sears avatar  avatar Valentin Plotkin avatar Bingo avatar  avatar menorki manil avatar ZB avatar  avatar  avatar zreevaz avatar  avatar lanye avatar Wei Deng avatar  avatar  avatar Yuan Zhiyi avatar HylaruCoder avatar Tang Jiasheng avatar Xiangyu Qi avatar  avatar  avatar  avatar 椅子卫士 avatar Ziqihuang avatar Zhiwei avatar tangzk avatar uang93 avatar Qu Dylan avatar Aiekick avatar  avatar  avatar Derrick Schultz (he/him) avatar Jae Jin avatar Holly Grimm avatar  avatar Tom Mulder avatar 赵伟松 avatar Arka Sadhu avatar saw avatar Tal Shiri avatar musicrainie avatar  avatar 爱可可-爱生活 avatar STYLIANOS IORDANIS avatar Ke Fang avatar Junho Kim avatar

Watchers

Shawn Presser avatar  avatar  avatar  avatar Hans Brouwer avatar lanye avatar Nazif BERAT avatar

stylegan_reimplementation's Issues

Breaking changes. What version of tensorflow should I use?

I was using tensorflow==1.15.0 on google collab

I had an error at models.py

Breaking changes happens all the time with tensorflow, so what version of TensorFlow should I use to replicate your dev environment?

Thank you.


dataset has 92 files
building graph
WARNING:tensorflow:From /content/stylegan_reimplementation/models.py:90: The name tf.keras.initializers.RandomNormal is deprecated. Please use tf.compat.v1.keras.initializers.RandomNormal instead.

WARNING:tensorflow:From /usr/local/lib/python3.6/dist-packages/tensorflow/python/keras/initializers.py:143: calling RandomNormal.__init__ (from tensorflow.python.ops.init_ops) with dtype is deprecated and will be removed in a future version.
Instructions for updating:
Call initializer instance with the dtype argument instead of passing it to the constructor
WARNING:tensorflow:From /content/stylegan_reimplementation/train.py:203: The name tf.train.AdamOptimizer is deprecated. Please use tf.compat.v1.train.AdamOptimizer instead.

Traceback (most recent call last):
  File "run.py", line 352, in <module>
    train.train(hps, files)
  File "/content/stylegan_reimplementation/train.py", line 460, in train
    num_shards=num_shards, shard_index=shard_index)
  File "/content/stylegan_reimplementation/train.py", line 196, in build_data_iterator
    label_list=label_list, num_shards=None, shard_index=None)
  File "/content/stylegan_reimplementation/data.py", line 77, in get_dataset
    dataset = make_raw_dataset(files)
TypeError: make_raw_dataset() missing 1 required positional argument: 'batch_size'
`
```

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.