pip3 install sparse_ae/
Для установки tensorflow-gpu
замените tensorflow
на tensorflow-gpu
в файле requirements.txt
python train_*.py
- обучение автокодировщика.
python compare_models.py
- строит гистограмму активаций обученных моделей.
plot_latent_code.py
- восстановление изображений из латентного представления.
partial_reconstruction.py
- восстановление изображений по top-k активных нейроно латентного слоя.
model.py
- создание моделей, KL-регуляризаторutils.py
- утилиты для подготовки данных и создания изображений.