-
Notifications
You must be signed in to change notification settings - Fork 3
/
dataset.py
33 lines (30 loc) · 872 Bytes
/
dataset.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
import numpy as np
image_size = [64, 128, 64]
trainset_cfg = dict(
data_dir='/SpineCTMR/train/**',
image_size=image_size,
aug_cfg=dict(prob=0.0,
translate_range=[3] * 3,
rotate_range=[np.pi / 60] * 3,
padding_mode='zeros'),
num_classes=26,
num_samples=2,
# check CacheDataset in MONAI
cache_rate=0.1,
num_workers=4,
)
valset_cfg = dict(
data_dir='/SpineCTMR/val/**',
image_size=image_size,
aug_cfg=dict(prob=0.0,
translate_range=[3] * 3,
rotate_range=[np.pi / 60] * 3,
padding_mode='zeros'),
num_classes=26,
num_samples=2,
# check CacheDataset in MONAI
cache_rate=0.1,
num_workers=4)
testset_cfg = dict(data_dir='/SpineCTMR/test/**',
cache_rate=0.1,
num_workers=4)