svqvae

Stacked Vector Quantized Variational Autoencoder for Hematologic Identification

read the report here: Project Report

Download the datasets to data/ folder before running

Name		Name	Last commit message	Last commit date
Latest commit History 6 Commits
checkpoints		checkpoints
data		data
models		models
.gitignore		.gitignore
README.md		README.md
step0-1_check_trainloader.py		step0-1_check_trainloader.py
step0-2_check_mask_loader.py		step0-2_check_mask_loader.py
step1-2_check_training_svqvae.py		step1-2_check_training_svqvae.py
step1_pretrain_svqvae.py		step1_pretrain_svqvae.py
step2_train_wbc_svqvae.py		step2_train_wbc_svqvae.py
step3_train_wbc_svqvae_with_mask.py		step3_train_wbc_svqvae_with_mask.py
step4-1_check_acc_SVQVAE.py		step4-1_check_acc_SVQVAE.py
step4-2_check_mask_acc_SVQVAE.py		step4-2_check_mask_acc_SVQVAE.py
step4-3_plot_loss.py		step4-3_plot_loss.py