Data Preparing
- Access to the synapse multi-organ dataset:
- Sign up in the official Synapse website and download the dataset. Convert them to numpy format, clip the images within [-125, 275], normalize each 3D image to [0, 1], and extract 2D slices from 3D volume for training cases while keeping the 3D volume in h5 format for testing cases.
- You can also send an Email directly to jienengchen01 AT gmail.com to request the preprocessed data for reproduction.
- The directory structure of the whole project is as follows:
.
βββ TransUNet
β βββdatasets
β β βββ dataset_*.py
β βββtrain.py
β βββtest.py
β βββ...
βββ model
β βββ vit_checkpoint
β βββ imagenet21k
β βββ R50+ViT-B_16.npz
β βββ *.npz
βββ data
βββSynapse
βββ test_vol_h5
β βββ case0001.npy.h5
β βββ *.npy.h5
βββ train_npz
βββ case0005_slice000.npz
βββ *.npz