├── .gitignore ├── LICENSE ├── README.md ├── bin ├── download_crello.sh ├── generate_crello_document.sh ├── generate_crello_image.sh ├── generate_rico.sh ├── train_canvasvae.sh └── train_pixelvae.sh ├── data └── .gitignore ├── docs ├── crello-dataset.md ├── interpolation.svg └── preview.svg ├── notebooks ├── crello-analysis.ipynb └── rico-analysis.ipynb ├── requirements.txt └── src ├── canvas-vae ├── README.md ├── canvasvae │ ├── __init__.py │ ├── __main__.py │ ├── data │ │ ├── __init__.py │ │ ├── component_legend.json │ │ ├── crello-document-spec.yml │ │ ├── rico-spec.yml │ │ └── spec.py │ ├── helpers │ │ ├── __init__.py │ │ ├── callbacks.py │ │ ├── random_eval.py │ │ ├── retrieve.py │ │ └── svg.py │ ├── main.py │ ├── models │ │ ├── __init__.py │ │ ├── decoder.py │ │ ├── discretizer.py │ │ ├── encoder.py │ │ ├── mask.py │ │ ├── metrics.py │ │ ├── transformer.py │ │ ├── utils.py │ │ └── vae.py │ └── train.py └── setup.py ├── pixel-vae ├── README.md ├── pixelvae │ ├── __init__.py │ ├── __main__.py │ ├── data.py │ ├── main.py │ ├── model.py │ ├── resnet_decoder.py │ └── train.py └── setup.py └── preprocess ├── README.md ├── preprocess ├── __init__.py ├── __main__.py ├── helpers │ ├── __init__.py │ ├── feature.py │ └── hash.py ├── main.py └── transforms │ ├── __init__.py │ ├── count.py │ ├── generate_crello_document.py │ ├── generate_crello_image.py │ ├── generate_rico.py │ └── vocabulary.py └── setup.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/README.md -------------------------------------------------------------------------------- /bin/download_crello.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/bin/download_crello.sh -------------------------------------------------------------------------------- /bin/generate_crello_document.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/bin/generate_crello_document.sh -------------------------------------------------------------------------------- /bin/generate_crello_image.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/bin/generate_crello_image.sh -------------------------------------------------------------------------------- /bin/generate_rico.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/bin/generate_rico.sh -------------------------------------------------------------------------------- /bin/train_canvasvae.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/bin/train_canvasvae.sh -------------------------------------------------------------------------------- /bin/train_pixelvae.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/bin/train_pixelvae.sh -------------------------------------------------------------------------------- /data/.gitignore: -------------------------------------------------------------------------------- 1 | * 2 | -------------------------------------------------------------------------------- /docs/crello-dataset.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/docs/crello-dataset.md -------------------------------------------------------------------------------- /docs/interpolation.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/docs/interpolation.svg -------------------------------------------------------------------------------- /docs/preview.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/docs/preview.svg -------------------------------------------------------------------------------- /notebooks/crello-analysis.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/notebooks/crello-analysis.ipynb -------------------------------------------------------------------------------- /notebooks/rico-analysis.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/notebooks/rico-analysis.ipynb -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/requirements.txt -------------------------------------------------------------------------------- /src/canvas-vae/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/src/canvas-vae/README.md -------------------------------------------------------------------------------- /src/canvas-vae/canvasvae/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/src/canvas-vae/canvasvae/__init__.py -------------------------------------------------------------------------------- /src/canvas-vae/canvasvae/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/src/canvas-vae/canvasvae/__main__.py -------------------------------------------------------------------------------- /src/canvas-vae/canvasvae/data/__init__.py: -------------------------------------------------------------------------------- 1 | from .spec import DataSpec 2 | -------------------------------------------------------------------------------- /src/canvas-vae/canvasvae/data/component_legend.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/src/canvas-vae/canvasvae/data/component_legend.json -------------------------------------------------------------------------------- /src/canvas-vae/canvasvae/data/crello-document-spec.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/src/canvas-vae/canvasvae/data/crello-document-spec.yml -------------------------------------------------------------------------------- /src/canvas-vae/canvasvae/data/rico-spec.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/src/canvas-vae/canvasvae/data/rico-spec.yml -------------------------------------------------------------------------------- /src/canvas-vae/canvasvae/data/spec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/src/canvas-vae/canvasvae/data/spec.py -------------------------------------------------------------------------------- /src/canvas-vae/canvasvae/helpers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/canvas-vae/canvasvae/helpers/callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/src/canvas-vae/canvasvae/helpers/callbacks.py -------------------------------------------------------------------------------- /src/canvas-vae/canvasvae/helpers/random_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/src/canvas-vae/canvasvae/helpers/random_eval.py -------------------------------------------------------------------------------- /src/canvas-vae/canvasvae/helpers/retrieve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/src/canvas-vae/canvasvae/helpers/retrieve.py -------------------------------------------------------------------------------- /src/canvas-vae/canvasvae/helpers/svg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/src/canvas-vae/canvasvae/helpers/svg.py -------------------------------------------------------------------------------- /src/canvas-vae/canvasvae/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/src/canvas-vae/canvasvae/main.py -------------------------------------------------------------------------------- /src/canvas-vae/canvasvae/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/canvas-vae/canvasvae/models/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/src/canvas-vae/canvasvae/models/decoder.py -------------------------------------------------------------------------------- /src/canvas-vae/canvasvae/models/discretizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/src/canvas-vae/canvasvae/models/discretizer.py -------------------------------------------------------------------------------- /src/canvas-vae/canvasvae/models/encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/src/canvas-vae/canvasvae/models/encoder.py -------------------------------------------------------------------------------- /src/canvas-vae/canvasvae/models/mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/src/canvas-vae/canvasvae/models/mask.py -------------------------------------------------------------------------------- /src/canvas-vae/canvasvae/models/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/src/canvas-vae/canvasvae/models/metrics.py -------------------------------------------------------------------------------- /src/canvas-vae/canvasvae/models/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/src/canvas-vae/canvasvae/models/transformer.py -------------------------------------------------------------------------------- /src/canvas-vae/canvasvae/models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/src/canvas-vae/canvasvae/models/utils.py -------------------------------------------------------------------------------- /src/canvas-vae/canvasvae/models/vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/src/canvas-vae/canvasvae/models/vae.py -------------------------------------------------------------------------------- /src/canvas-vae/canvasvae/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/src/canvas-vae/canvasvae/train.py -------------------------------------------------------------------------------- /src/canvas-vae/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/src/canvas-vae/setup.py -------------------------------------------------------------------------------- /src/pixel-vae/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/src/pixel-vae/README.md -------------------------------------------------------------------------------- /src/pixel-vae/pixelvae/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/pixel-vae/pixelvae/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/src/pixel-vae/pixelvae/__main__.py -------------------------------------------------------------------------------- /src/pixel-vae/pixelvae/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/src/pixel-vae/pixelvae/data.py -------------------------------------------------------------------------------- /src/pixel-vae/pixelvae/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/src/pixel-vae/pixelvae/main.py -------------------------------------------------------------------------------- /src/pixel-vae/pixelvae/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/src/pixel-vae/pixelvae/model.py -------------------------------------------------------------------------------- /src/pixel-vae/pixelvae/resnet_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/src/pixel-vae/pixelvae/resnet_decoder.py -------------------------------------------------------------------------------- /src/pixel-vae/pixelvae/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/src/pixel-vae/pixelvae/train.py -------------------------------------------------------------------------------- /src/pixel-vae/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/src/pixel-vae/setup.py -------------------------------------------------------------------------------- /src/preprocess/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/src/preprocess/README.md -------------------------------------------------------------------------------- /src/preprocess/preprocess/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/preprocess/preprocess/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/src/preprocess/preprocess/__main__.py -------------------------------------------------------------------------------- /src/preprocess/preprocess/helpers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/preprocess/preprocess/helpers/feature.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/src/preprocess/preprocess/helpers/feature.py -------------------------------------------------------------------------------- /src/preprocess/preprocess/helpers/hash.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/src/preprocess/preprocess/helpers/hash.py -------------------------------------------------------------------------------- /src/preprocess/preprocess/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/src/preprocess/preprocess/main.py -------------------------------------------------------------------------------- /src/preprocess/preprocess/transforms/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/src/preprocess/preprocess/transforms/__init__.py -------------------------------------------------------------------------------- /src/preprocess/preprocess/transforms/count.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/src/preprocess/preprocess/transforms/count.py -------------------------------------------------------------------------------- /src/preprocess/preprocess/transforms/generate_crello_document.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/src/preprocess/preprocess/transforms/generate_crello_document.py -------------------------------------------------------------------------------- /src/preprocess/preprocess/transforms/generate_crello_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/src/preprocess/preprocess/transforms/generate_crello_image.py -------------------------------------------------------------------------------- /src/preprocess/preprocess/transforms/generate_rico.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/src/preprocess/preprocess/transforms/generate_rico.py -------------------------------------------------------------------------------- /src/preprocess/preprocess/transforms/vocabulary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/src/preprocess/preprocess/transforms/vocabulary.py -------------------------------------------------------------------------------- /src/preprocess/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CyberAgentAILab/canvas-vae/HEAD/src/preprocess/setup.py --------------------------------------------------------------------------------