Skip to content

Instantly share code, notes, and snippets.

@omarsar

omarsar/mt_split_data.py

Created October 15, 2018 09:08
Show Gist options
  • Save omarsar/ad46d4be9759fe7292cfeb49de58c1ec to your computer and use it in GitHub Desktop.
Save omarsar/ad46d4be9759fe7292cfeb49de58c1ec to your computer and use it in GitHub Desktop.
Download ZIP
Raw
mt_split_data.py
### training dataset with 80/20 split
TRAIN_ROOT_DIR_GMCHALLENGE = "/gdrive/My Drive/DAIR RESOURCES/PyTorch/medical_imaging/train/"
gmdataset_train = mt_datasets.SCGMChallenge2DTrain(root_dir=TRAIN_ROOT_DIR_GMCHALLENGE,
subj_ids=range(1, 9),
transform=train_transform,
slice_filter_fn=mt_filters.SliceFilter())
gmdataset_val = mt_datasets.SCGMChallenge2DTrain(root_dir=TRAIN_ROOT_DIR_GMCHALLENGE,
subj_ids=range(9, 11),
transform=val_transform)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment