first commit

ssd/config/hat-512.yaml

@@ -0,0 +1,30 @@
+MODEL:
+  NUM_CLASSES: 21
+  BACKBONE:
+    OUT_CHANNELS: (512, 1024, 512, 256, 256, 256, 256)
+  PRIORS:
+    FEATURE_MAPS: [64, 32, 16, 8, 4, 2, 1]
+    STRIDES: [8, 16, 32, 64, 128, 256, 512]
+    MIN_SIZES: [35.84, 76.8, 153.6, 230.4, 307.2, 384.0, 460.8]
+    MAX_SIZES: [76.8, 153.6, 230.4, 307.2, 384.0, 460.8, 537.65]
+    ASPECT_RATIOS: [[2], [2, 3], [2, 3], [2, 3], [2, 3], [2], [2]]
+    BOXES_PER_LOCATION: [4, 6, 6, 6, 6, 4, 4]
+INPUT:
+  IMAGE_SIZE: 512
+DATASETS:
+  #TRAIN: ("voc_2007_birdtrain", "voc_2007_birdval")
+  #TRAIN: ("voc_2007_trainval", )
+  #TEST: ("voc_2007_test", )
+  #TEST: ("vocbird_2019_test", )
+  TRAIN: ("voc_3_train",)
+  TEST: ("voc_3_test", )
+  INF: ("inference_hatcher",)
+SOLVER:
+  MAX_ITER: 175000
+
+  LR_STEPS: [80000, 100000]
+  GAMMA: 0.1
+  BATCH_SIZE: 16
+  LR: 1e-5
+#1e-3
+OUTPUT_DIR: '/data1/Hatcher/model/data_MPKFNet/weigth/detection/vgg_ssd512_voc0712VOC3/'