Ready to use implementations of state-of-the-art generative models in TensorFlow 2
Installation
Dependencies
tf-gen-models requires:
- Python (>= 3.7, < 3.10)
- TensorFlow (>= 2.5)
- Matplotlib (>= 3.4)
- Pillow (>= 8.0)
The tf-gen-models package is built upon TensorFlow 2. See the TensorFlow install guide for the pip package while, to enable GPU support, the use Docker container is recommended. Alternatively, GPU-enabled TensorFlow can be easily installed using the tensorflow-gpu package on conda-forge.
User installation
If you already have a working installation of TensorFlow 2 (preferably with the GPU support enabled), the easiest way to install tf-gen-models is using pip:
pip install tf-gen-models
Available generative models
| Generative models | Implementation | Notebooks | Trends |
|---|---|---|---|
| GAN | ✔️ | 🛠️ | |
| VAE | ❌ | ❌ | |
| Norm Flow | ❌ | ❌ |
Generative Adversarial Networks
| Algorithms | Implementation | Conditioning* | Notebooks | Paper |
|---|---|---|---|---|
GAN |
✔️ | 🛠️ | ✔️ | arXiv:1406.2661 |
BceGAN |
✔️ | ❌ | ✔️ | |
WGAN |
✔️ | ❌ | ✔️ | arXiv:1701.07875 |
WGAN_GP |
✔️ | ❌ | ✔️ | arXiv:1704.00028 |
CramerGAN |
✔️ | ❌ | ✔️ | arXiv:1705.10743 |
WGAN_ALP |
✔️ | ❌ | 🛠️ | arXiv:1907.05681 |
*Referring to the conditional version of GANs proposed in arXiv:1411.1784.
Variational Autoencoder
Planned for release v0.2.0
Normalizing Flows
Planned for release v0.2.0
Jupyter notebooks
License
1 maintainer
Release files for tf-gen-models 0.0.12
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| tf_gen_models-0.0.12-py3-none-any.whl | Python 3 | none | any | Details |
Release files / tf_gen_models-0.0.12-py3-none-any.whl
| Download URL | tf_gen_models-0.0.12-py3-none-any.whl |
|---|---|
| Size | 19.3 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
7db6796b600604eb9e70abf018edb0b8c26e2eb88db7e9415d6920dd0c165150
|
|
BLAKE2b-256 checksum How to use checksums |
530bd58110898b216fcf99f3728ff4b93f09675dc2572ac292a8527bb25d096d
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/3.7.1 importlib_metadata/4.8.2 pkginfo/1.8.2 requests/2.26.0 requests-toolbelt/0.9.1 tqdm/4.62.3 CPython/3.8.12
|
