README.md 1.3 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
# Structure of the repository
- There are two directories in the **Individual_Project** directory:
    - **data**: this directory stores the dataset used in this project, such as the Morpho-MNIST dataset. The cardiac anatomy dataset cannot be uploaded because of the privacy policy.
    - **model**: this directory stores the notebooks and output of the project.

- There are three directories in the **Individual_Project/model/Output** Directory:
    - **MNIST**: this directory stores the trained VQ-VAE-2 and PixelSNAIL models on the Morpho-MNIST dataset, and TensorBoard logs.
    - **Segmentation_2D**: this directory stores the trained VQ-VAE-2 and PixelSNAIL models on the 2D cardiac segmentation dataset, and TensorBoard logs.
    - **Segmentation_3D**: this directory stores the trained VQ-VAE-2 models on the 3D cardiac segmentation dataset, and TensorBoard logs.

song, hongkang's avatar
song, hongkang committed
11
- There are three VQ-VAE-2 models in the **Individual_Project/model/Output/Segmentation_3D/models** directory:
12
13
14
15
    - **best.pt**: trained VQ-VAE-2 model with codebook size 512×64 and using random restarts method
    - **best_32_16_no_restart.pt**: trained VQ-VAE-2 model with codebook size 32×16 and not using random restarts method
    - **best_32_116_restart.pt**: trained VQ-VAE-2 model with codebook size 32×16 and using random restarts method