|
--- |
|
title: Ae Gen |
|
emoji: 💻 |
|
colorFrom: yellow |
|
colorTo: pink |
|
sdk: gradio |
|
sdk_version: 3.16.2 |
|
app_file: app.py |
|
pinned: false |
|
license: mit |
|
--- |
|
|
|
- ConvAE model (from [`Digits that are not: Generating new types through deep neural nets`](https://arxiv.org/pdf/1606.04345.pdf)) |
|
- DeepConvAE model (from [here](https://tel.archives-ouvertes.fr/tel-01838272/file/75406_CHERTI_2018_diffusion.pdf), Section 10.1 with `L=3`) |
|
- Dense K-Sparse model (from [`Out-of-class novelty generation`](https://openreview.net/forum?id=r1QXQkSYg)) |
|
|
|
These models were trained on MNIST only (digits), but were found to generate new kinds of symbols, see the references for more details. |
|
|
|
 |
|
|
|
# Install requirements |
|
|
|
`pip install -r requirements.txt` |
|
|
|
# Download models |
|
|
|
```bash |
|
git lfs pull |
|
``` |
|
|
|
# Training |
|
|
|
`python cli.py train --dataset=mnist --folder=convae --model=convae` |
|
|
|
`python cli.py train --dataset=mnist --folder=deep_convae --model=deep_convae` |
|
|
|
# Generate samples |
|
|
|
```bash |
|
python cli.py test --model-path=convae.th --nb-generate=100 --folder=convae |
|
``` |
|
|
|
```bash |
|
python cli.py test --model-path=deep_convae.th --nb-generate=100 --folder=deep_convae |
|
``` |
|
|
|
```bash |
|
python cli.py test --model-path=fc_sparse.th --nb-generate=100 --folder=deep_convae |
|
``` |
|
|
