-
Notifications
You must be signed in to change notification settings - Fork 1
/
configs.py
36 lines (30 loc) · 1.05 KB
/
configs.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
34
35
36
# some directory for output the results:
OUTPUT_PATH = '/D:/Academic/MSC/Research/Code/output'
# Resource path which contains: annotations.csv, candidates.csv,
# and subdirectories containing .mhd files.
# This is the directory structure needed to run the code:
# (The code will use all .mhd and .raw files inside subdirectories which their name is in annotations or candidates)
'''
[RESOURCES_PATH]/
annotations.csv
candidates.csv
subset0/
*.mhd
*.raw
subset1/
*.mhd
*.raw
my_custom_subset/
*.mhd
*.raw
'''
RESOURCES_PATH = '/D:/Academic/MSC/Research/Data/input'
PADDING_FOR_LOCALIZATION = 10
BLOCK_SIZE = 128
COORDS_CUBE_SIZE = 32
TARGET_SHAPE = (COORDS_CUBE_SIZE, COORDS_CUBE_SIZE, COORDS_CUBE_SIZE, 3, 5)
COORDS_SHAPE = (3, COORDS_CUBE_SIZE, COORDS_CUBE_SIZE, COORDS_CUBE_SIZE)
ANCHOR_SIZES = [10, 30, 60]
VAL_PCT = 0.2
TOTAL_EPOCHS = 100
DEFAULT_LR = 0.01