Commit 28f98693 authored by song, hongkang's avatar song, hongkang
Browse files

Update README.md

parent 50b48c13
......@@ -8,7 +8,7 @@
- **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.
- There are three VQ-VAE-2 models in the **Individual_Project/model/Output/Segmentation_3D** directory:
- There are three VQ-VAE-2 models in the **Individual_Project/model/Output/Segmentation_3D/models** directory:
- **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
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment