Newer
Older
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
MODEL:
BACKBONE:
NAME: 'vgg16'
RPN:
ANCHOR_SIZES: (32, 64, 128, 256, 512)
ROI_BOX_HEAD:
NUM_CLASSES: 9
BOX_PREDICTOR: 'vgg16_predictor'
POOL_TYPE: 'align'
DATASETS:
TRAINS: ['cityscapes_train']
TARGETS: ['foggy_cityscapes_train_0.02']
TESTS: ['foggy_cityscapes_val_0.02']
INPUT:
TRANSFORMS_TRAIN:
- name: 'random_flip'
- name: 'resize'
min_size: 600
- name: 'normalize'
mean: [0.5, 0.5, 0.5]
std: [0.5, 0.5, 0.5]
to_01: True
- name: 'collect'
TRANSFORMS_TEST:
- name: 'resize'
min_size: 600
- name: 'normalize'
mean: [0.5, 0.5, 0.5]
std: [0.5, 0.5, 0.5]
to_01: True
- name: 'collect'
SOLVER:
EPOCHS: 25
STEPS: (16, 22)
LR: 1e-5
BATCH_SIZE: 1
TEST:
EVAL_TYPES: ['voc']
WORK_DIR: './work_dir/adv_cityscapes_2_foggy'