Skip to content
baseline_vgg16_cityscapes_2_foggy.yaml 707 B
Newer Older
李聪聪's avatar
李聪聪 committed
MODEL:
  BACKBONE:
    NAME: 'vgg16'
  ROI_BOX_HEAD:
    NUM_CLASSES: 9
    BOX_PREDICTOR: 'vgg16_predictor'
    POOL_TYPE: 'pooling'
DATASETS:
  TRAINS: ['cityscapes_train']
  TESTS: ['foggy_cityscapes_val_0.02']
INPUT:
  TRANSFORMS_TRAIN:
李聪聪's avatar
李聪聪 committed
    - 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
李聪聪's avatar
李聪聪 committed
  TRANSFORMS_TEST:
李聪聪's avatar
李聪聪 committed
    - name: 'resize'
      min_size: 600
    - name: 'normalize'
      mean: [0.5, 0.5, 0.5]
      std: [0.5, 0.5, 0.5]
      to_01: True
李聪聪's avatar
李聪聪 committed
SOLVER:
  EPOCHS: 25
  STEPS: (16, 22)
  LR: 1e-5
  BATCH_SIZE: 1
TEST:
  EVAL_TYPES: ['voc']

WORK_DIR: './work_dir/baseline_cityscapes_2_foggy'