Add vehicle configs

configs/VERIWild/bagtricks_R50-ibn.yml

@@ -0,0 +1,36 @@
+_BASE_: "../Base-bagtricks.yml"
+
+INPUT:
+  SIZE_TRAIN: [256, 256]
+  SIZE_TEST: [256, 256]
+
+MODEL:
+  BACKBONE:
+    WITH_IBN: True
+    PRETRAIN_PATH: '/export2/home/zjk/pretrain_models/resnet50_ibn_a.pth.tar'
+  HEADS:
+    NUM_CLASSES: 30671
+    POOL_LAYER: gempool
+  LOSSES:
+    TRI:
+      HARD_MINING: False
+      MARGIN: 0.0
+
+DATASETS:
+  NAMES: ("VeRiWild",)
+  TESTS: ("SmallVeRiWild", "MediumVeRiWild", "LargeVeRiWild",)
+
+SOLVER:
+  IMS_PER_BATCH: 512
+  MAX_ITER: 64800
+  STEPS: [21600, 48600]
+  WARMUP_ITERS: 5400
+
+  LOG_PERIOD: 540
+  CHECKPOINT_PERIOD: 21600
+
+TEST:
+  EVAL_PERIOD: 5400
+  IMS_PER_BATCH: 1024
+
+OUTPUT_DIR: "logs/veriwild/bagtricks_R50-ibn_4gpu"

configs/VeRi/sbs_R50-ibn.yml

@@ -0,0 +1,37 @@
+_BASE_: "../Base-Strongerbaseline.yml"
+
+INPUT:
+  SIZE_TRAIN: [256, 256]
+  SIZE_TEST: [256, 256]
+
+MODEL:
+  BACKBONE:
+    WITH_IBN: True
+    PRETRAIN_PATH: "/export2/home/zjk/pretrain_models/resnet50_ibn_a.pth.tar"
+
+  HEADS:
+    NUM_CLASSES: 575
+
+SOLVER:
+  OPT: "SGD"
+  BASE_LR: 0.01
+  ETA_MIN_LR: 7.7e-5
+
+  IMS_PER_BATCH: 64
+  MAX_ITER: 35400
+  DELAY_ITERS: 17700
+  WARMUP_ITERS: 2950
+  FREEZE_ITERS: 2950
+
+  LOG_PERIOD: 590
+  CHECKPOINT_PERIOD: 11800
+
+DATASETS:
+  NAMES: ("VeRi",)
+  TESTS: ("VeRi",)
+
+TEST:
+  EVAL_PERIOD: 3540
+  IMS_PER_BATCH: 128
+
+OUTPUT_DIR: "logs/veri/sbs_R50-ibn"

configs/VehicleID/bagtricks_R50-ibn.yml

@@ -0,0 +1,38 @@
+_BASE_: "../Base-bagtricks.yml"
+
+INPUT:
+  SIZE_TRAIN: [256, 256]
+  SIZE_TEST: [256, 256]
+
+MODEL:
+  BACKBONE:
+    WITH_IBN: True
+    PRETRAIN_PATH: '/export2/home/zjk/pretrain_models/resnet50_ibn_a.pth.tar'
+  HEADS:
+    NUM_CLASSES: 13164
+    POOL_LAYER: gempool
+  LOSSES:
+    TRI:
+      HARD_MINING: False
+      MARGIN: 0.0
+
+DATASETS:
+  NAMES: ("VehicleID",)
+  TESTS: ("SmallVehicleID", "MediumVehicleID", "LargeVehicleID",)
+
+SOLVER:
+  BIAS_LR_FACTOR: 1.
+
+  IMS_PER_BATCH: 512
+  MAX_ITER: 28600
+  STEPS: [8800, 19800]
+  WARMUP_ITERS: 2200
+
+  LOG_PERIOD: 220
+  CHECKPOINT_PERIOD: 8800
+
+TEST:
+  EVAL_PERIOD: 2200
+  IMS_PER_BATCH: 1024
+
+OUTPUT_DIR: "logs/vehicleid/bagtricks_R50-ibn_4gpu"