110-1-NTU-DBME5028-Histology-images-query
Final Project: Histology images query (unsupervised) Kaggle: https://www.kaggle.com/c/histology-images-query-competition/overview
Teem members:
f09921058 陳羿翔, r09942171 黃繼綸
Environment
OS : Ubuntu 16.04
Language: Python37 Torch: 21.08+
Introduction
Processing
Model (VAE + BYOL)
How to use
Download dataset in Kaggle
https://www.kaggle.com/c/histology-images-query-competition/data
$ kaggle competitions download -c histology-images-query-competition
Download the pretrained models
https://drive.google.com/drive/folders/1u2wWuPfb327AHGi8WRCTMiVxmIRPrUHg?usp=sharing
Training from scratch
$ python train.py -v --eta 10 --save_model_path path/to/save --data path/to/train
- -v: save vae result
- --eta: Ltotal = Ltask + \eta * Lreconstruction
- --lamda: Lconstruction = Lmse + \lamda * Lkl
Testing
$ python inference.py --data path/to/test