Skip to content
baseline_vgg16_cityscapes_2_foggy.yaml 804 B
Newer Older
李聪聪's avatar
李聪聪 committed
MODEL:
  BACKBONE:
    NAME: 'vgg16'
李聪聪's avatar
李聪聪 committed
  RPN:
    ANCHOR_SIZES: (32, 64, 128, 256, 512)
李聪聪's avatar
李聪聪 committed
  ROI_BOX_HEAD:
    NUM_CLASSES: 9
    BOX_PREDICTOR: 'vgg16_predictor'
李聪聪's avatar
李聪聪 committed
    POOL_TYPE: 'align'
李聪聪's avatar
李聪聪 committed
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
    - name: 'collect'
李聪聪'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
    - name: 'collect'
李聪聪's avatar
李聪聪 committed
SOLVER:
  EPOCHS: 25
  STEPS: (16, 22)
  LR: 1e-5
  BATCH_SIZE: 1
TEST:
  EVAL_TYPES: ['voc']

李聪聪's avatar
李聪聪 committed
WORK_DIR: './work_dir/baseline_cityscapes_2_foggy_align'