├── .gitignore ├── README.md ├── VAE_colab.ipynb ├── dataset.zip ├── dataset_scripts ├── image_utils.py ├── minifig-heads.csv └── scraper_bricklink.py ├── docs └── img │ ├── face-morph-1.png │ ├── face-morph-2.png │ ├── face-morph-3.png │ └── reconstruction.png ├── ml ├── constants.py ├── utils.py └── variational_autoencoder.py ├── requirements.txt └── trained_model ├── params.pkl └── weights.h5 /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iechevarria/lego-face-VAE/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iechevarria/lego-face-VAE/HEAD/README.md -------------------------------------------------------------------------------- /VAE_colab.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iechevarria/lego-face-VAE/HEAD/VAE_colab.ipynb -------------------------------------------------------------------------------- /dataset.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iechevarria/lego-face-VAE/HEAD/dataset.zip -------------------------------------------------------------------------------- /dataset_scripts/image_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iechevarria/lego-face-VAE/HEAD/dataset_scripts/image_utils.py -------------------------------------------------------------------------------- /dataset_scripts/minifig-heads.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iechevarria/lego-face-VAE/HEAD/dataset_scripts/minifig-heads.csv -------------------------------------------------------------------------------- /dataset_scripts/scraper_bricklink.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iechevarria/lego-face-VAE/HEAD/dataset_scripts/scraper_bricklink.py -------------------------------------------------------------------------------- /docs/img/face-morph-1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iechevarria/lego-face-VAE/HEAD/docs/img/face-morph-1.png -------------------------------------------------------------------------------- /docs/img/face-morph-2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iechevarria/lego-face-VAE/HEAD/docs/img/face-morph-2.png -------------------------------------------------------------------------------- /docs/img/face-morph-3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iechevarria/lego-face-VAE/HEAD/docs/img/face-morph-3.png -------------------------------------------------------------------------------- /docs/img/reconstruction.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iechevarria/lego-face-VAE/HEAD/docs/img/reconstruction.png -------------------------------------------------------------------------------- /ml/constants.py: -------------------------------------------------------------------------------- 1 | DEFAULT_MODEL_PATH = "model_output" 2 | -------------------------------------------------------------------------------- /ml/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iechevarria/lego-face-VAE/HEAD/ml/utils.py -------------------------------------------------------------------------------- /ml/variational_autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iechevarria/lego-face-VAE/HEAD/ml/variational_autoencoder.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iechevarria/lego-face-VAE/HEAD/requirements.txt -------------------------------------------------------------------------------- /trained_model/params.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iechevarria/lego-face-VAE/HEAD/trained_model/params.pkl -------------------------------------------------------------------------------- /trained_model/weights.h5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/iechevarria/lego-face-VAE/HEAD/trained_model/weights.h5 --------------------------------------------------------------------------------