完成训练模块的转移

deep_sort/configs/deep_sort.yaml

@@ -0,0 +1,10 @@
+DEEPSORT:
+  REID_CKPT: "./deep_sort/deep/checkpoint/ckpt.t7"
+  MAX_DIST: 0.2
+  MIN_CONFIDENCE: 0.5
+  NMS_MAX_OVERLAP: 0.5
+  MAX_IOU_DISTANCE: 0.7
+  MAX_AGE: 70
+  N_INIT: 3
+  NN_BUDGET: 100
+  

deep_sort/configs/fastreid.yaml

@@ -0,0 +1,3 @@
+FASTREID:
+  CFG: "thirdparty/fast-reid/configs/Market1501/bagtricks_R50.yml"
+  CHECKPOINT: "deep_sort/deep/checkpoint/market_bot_R50.pth"

deep_sort/configs/mask_rcnn.yaml

@@ -0,0 +1,6 @@
+MASKRCNN:
+  LABEL: "./coco_classes.json"
+  WEIGHT: "./detector/Mask_RCNN/save_weights/maskrcnn_resnet50_fpn_coco.pth"
+
+  NUM_CLASSES: 90
+  BOX_THRESH: 0.5

deep_sort/configs/mmdet.yaml

@@ -0,0 +1,5 @@
+MMDET:
+  CFG: "thirdparty/mmdetection/configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py"
+  CHECKPOINT: "detector/MMDet/weight/faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth"
+
+  SCORE_THRESH: 0.5