Skip to content

Library about construction helper for Generative models e.g. Flow-based Model with Tensorflow 2.x.

Notifications You must be signed in to change notification settings

MokkeMeguru/TFGENZOO

Repository files navigation

TFGENZOO (Generative Model x Tensorflow 2.x)

img img img img img

What’s this repository?

This is a repository for some researcher to build some Generative models using Tensorflow 2.x.

I NEED YOUR HELP(please let me know about formula, implementation and anything you worried)

img

Zen of this repository

We don't want to need flexible architectures.
We need strict definitions for shapes, parameters, and formulas.
We should Implement correct codes with well-documented(tested).

How to use?

By Install

  • pipenv

    pipenv install TFGENZOO==1.2.5
    
  • pip

    pip install TFGENZOO==1.2.5
    

Source build for development

  1. clone this repository (If you want to do it, I will push this repository to PYPI)
  2. build this repository docker-compose build
  3. run the environment sh run_script.sh
  4. connect it via VSCode or Emacs or vi or anything.

Examples

Documents

https://mokkemeguru.github.io/TFGENZOO/

Roadmap

  • Flow-based Model Architecture (RealNVP, Glow)
  • i-ResNet Model Architecture (i-ResNet, i-RevNet)
  • GANs Model Architecture (GANs)

Remarkable Backlog

Whole backlog is here

News [2020/6/16]

New training results Oxford-flower102 with only 8 hours! (Quadro P6000 x 1)

data NLL(test) epoch pretrained
Oxford-flower102 4.590211391448975 1024 ---

img

see more detail, you can see my internship’s report (Japanese only, if you need translated version, please contact me.)

News [2020/7/11]

Add some tutorial into ./tutorial

News [2021/3/30]

I wrote the master's paper about japanese text style transfer. "AutoEncoder に基づく半教師あり和文スタ イル変換" https://drive.google.com/file/d/1KtkLZi6PUvL7msAqbg_KRdEC0pmmpbhf/view?usp=sharing

Contact

MokkeMeguru (@MokkeMeguru): DM or Mention Please (in Any language).