-
Notifications
You must be signed in to change notification settings - Fork 1.7k
/
abdomen.yml
40 lines (37 loc) · 942 Bytes
/
abdomen.yml
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
34
35
36
37
38
39
40
_base_: '../_base_/global_configs.yml'
train_dataset:
type: Abdomen
dataset_root: abdomen/abdomen_phase0
result_dir: abdomen/abdomen_phase1
transforms:
- type: RandomFlip3D
flip_axis: [1, 2]
- type: RandomQuarterTurn3D
rotate_planes: [[ 1, 2 ]]
- type: RandomRotation3D
degrees: 20
rotate_planes: [[1, 2]]
- type: Resize3D
size: [1, 224, 224]
mode: train
num_classes: 9
val_dataset:
type: Abdomen
dataset_root: abdomen/abdomen_phase0
result_dir: abdomen/abdomen_phase1
num_classes: 9
transforms:
- type: Resize3D
size: [1, 224, 224]
mode: val
dataset_json_path: data/abdomen/abdomen_raw/dataset.json
test_dataset:
type: Abdomen
dataset_root: abdomen/abdomen_phase0
result_dir: abdomen/abdomen_phase1
num_classes: 9
transforms:
- type: Resize3D
size: [1, 224, 224]
mode: val
dataset_json_path: data/abdomen/abdomen_raw/dataset.json