Commit 6f85991f authored by 李聪聪's avatar 李聪聪
Browse files

v3

parent 76b93b29

adversarial_train.py

+3 −599

File changed.

Preview size limit exceeded, changes collapsed.

adversarial_train_deprecated .py

0 → 100644
+878 −0

File added.

Preview size limit exceeded, changes collapsed.

configs/adv_vgg16_cityscapes_2_foggy.yaml

+0 −10
Original line number Diff line number Diff line
@@ -16,16 +16,6 @@ ADV: 
      pool_type: 'avg'

      loss_weight: 1.0

      window_sizes: [3, 9, 15, 21, -1]

  METHOD: 'netvlad'

  LOSS_FUNC: 'cross_entropy'

  NETVLAD:

    WEIGHTED: True

  SELECTIVE:

    METHOD: 'meta'

  WINDOWS:

    START_SIZE: 3

    STOP_SIZE: -1

    NUM_WINDOWS: 5

DATASETS:

  TRAINS: ['cityscapes_train']

  TARGETS: ['foggy_cityscapes_train_0.02']

configs/adv_vgg16_kitti_2_cityscapes.yaml

+13 −3
Original line number Diff line number Diff line
@@ -6,9 +6,19 @@ MODEL: 
    BOX_PREDICTOR: 'vgg16_predictor'

    POOL_TYPE: 'pooling'

ADV:

  LOSS_WEIGHT: 1.0

  FOCAL_LOSS_GAMMA: 2

  LOSS_FUNC: 'focal_loss'

  LAYERS: [True, False, True]

  DIS_MODEL:

    - in_channels: 256

      func_name: 'cross_entropy'

      pool_type: 'avg'

      loss_weight: 1

      window_sizes: [3, 9]

    - in_channels: 512

      func_name: 'focal_loss'

      focal_loss_gamma: 5

      pool_type: 'avg'

      loss_weight: 1

      window_sizes: [3, 9, 15, 21, -1]

DATASETS:

  TRAINS: ['kitti_train']

  TARGETS: ['cityscapes_car_train']