This repository has been archived by the owner on Nov 16, 2023. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 138
/
hrnet.yaml
105 lines (97 loc) · 1.88 KB
/
hrnet.yaml
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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
CUDNN:
BENCHMARK: true
DETERMINISTIC: false
ENABLED: true
GPUS: (0,)
OUTPUT_DIR: 'output'
LOG_DIR: 'log'
WORKERS: 4
PRINT_FREQ: 10
LOG_CONFIG: logging.conf
SEED: 2019
OPENCV_BORDER_CONSTANT: 0
DATASET:
NUM_CLASSES: 6
ROOT: "/home/username/data/dutch/data"
CLASS_WEIGHTS: [0.7151, 0.8811, 0.5156, 0.9346, 0.9683, 0.9852]
MIN: -1
MAX: 1
MODEL:
NAME: seg_hrnet
IN_CHANNELS: 3
PRETRAINED: "<ImageNet pretrained model path - please see main README>"
EXTRA:
FINAL_CONV_KERNEL: 1
STAGE2:
NUM_MODULES: 1
NUM_BRANCHES: 2
BLOCK: BASIC
NUM_BLOCKS:
- 4
- 4
NUM_CHANNELS:
- 48
- 96
FUSE_METHOD: SUM
STAGE3:
NUM_MODULES: 4
NUM_BRANCHES: 3
BLOCK: BASIC
NUM_BLOCKS:
- 4
- 4
- 4
NUM_CHANNELS:
- 48
- 96
- 192
FUSE_METHOD: SUM
STAGE4:
NUM_MODULES: 3
NUM_BRANCHES: 4
BLOCK: BASIC
NUM_BLOCKS:
- 4
- 4
- 4
- 4
NUM_CHANNELS:
- 48
- 96
- 192
- 384
FUSE_METHOD: SUM
TRAIN:
BATCH_SIZE_PER_GPU: 16
BEGIN_EPOCH: 0
END_EPOCH: 300
MIN_LR: 0.001
MAX_LR: 0.02
MOMENTUM: 0.9
WEIGHT_DECAY: 0.0001
SNAPSHOTS: 5
AUGMENTATION: True
DEPTH: "section" # Options are: none, patch, and section
STRIDE: 50
PATCH_SIZE: 100
AUGMENTATIONS:
RESIZE:
HEIGHT: 200
WIDTH: 200
PAD:
HEIGHT: 256
WIDTH: 256
MEAN: 0.0009997 # 0.0009996710808862074
STD: 0.20977 # 0.20976548783479299
MODEL_DIR: "models"
VALIDATION:
BATCH_SIZE_PER_GPU: 128
TEST:
MODEL_PATH: "/home/username/seismic-deeplearning/docker/hrnetv2_w48_imagenet_pretrained.pth"
TEST_STRIDE: 10
SPLIT: 'Both' # Can be Both, Test1, Test2
INLINE: True
CROSSLINE: True
POST_PROCESSING:
SIZE: 128 #
CROP_PIXELS: 14 # Number of pixels to crop top, bottom, left and right