dnt 0.2.4__py3-none-any.whl → 0.3.1.3__py3-none-any.whl

dnt/shared/util.py

@@ -18,19 +18,32 @@ import cv2
 
 visdrone_classes = {'car': 4, 'bus': 9, 'truck': 6, 'pedestrian': 1, 'van': 5}
 
-def load_class_dict(name_file=None):
+def load_class_dict(name_file:str=None)->dict:
     if not name_file:
         lib_root = pathlib.Path(__file__).resolve().parents[1]
         data_path = os.path.join(lib_root, 'shared/data')
         name_file = os.path.join(data_path, 'coco.names')
 
-        print(name_file)
-        input('...')
-
     class_dict = pd.read_csv(name_file,header=None).to_dict()
     class_dict_reverse = dict(map(reversed, class_dict.items()))
+
+    print(class_dict_reverse)
+    input('...')
     return class_dict_reverse
 
+def load_classes(name_file=None)->list:
+    if not name_file:
+        lib_root = pathlib.Path(__file__).resolve().parents[1]
+        data_path = os.path.join(lib_root, 'shared/data')
+        name_file = os.path.join(data_path, 'coco.names')
+
+    my_file = open(name_file, "r") 
+    data = my_file.read() 
+    results = data.split("\n")
+    my_file.close()
+
+    return results
+
 def load_mot16(detections, nms_overlap_thresh=None, with_classes=True, nms_per_class=False, class_filename=""):
     """
     Loads detections stored in a mot-challenge like formatted CSV or numpy array (fieldNames = ['frame', 'id', 'x', 'y',

dnt/third_party/fast-reid/__init__.py

@@ -0,0 +1 @@
+from fast_reid_interfece import get_cfg, FastReIDInterface, preprocess, postprocess

dnt/third_party/fast-reid/configs/Base-AGW.yml

@@ -0,0 +1,19 @@
+_BASE_: Base-bagtricks.yml
+
+MODEL:
+  BACKBONE:
+    WITH_NL: True
+
+  HEADS:
+    POOL_LAYER: GeneralizedMeanPooling
+
+  LOSSES:
+    NAME: ("CrossEntropyLoss", "TripletLoss")
+    CE:
+      EPSILON: 0.1
+      SCALE: 1.0
+
+    TRI:
+      MARGIN: 0.0
+      HARD_MINING: False
+      SCALE: 1.0

dnt/third_party/fast-reid/configs/Base-MGN.yml

@@ -0,0 +1,12 @@
+_BASE_: Base-SBS.yml
+
+MODEL:
+  META_ARCHITECTURE: MGN
+
+  FREEZE_LAYERS: [backbone, b1, b2, b3,]
+
+  BACKBONE:
+    WITH_NL: False
+
+  HEADS:
+    EMBEDDING_DIM: 256

dnt/third_party/fast-reid/configs/Base-SBS.yml

@@ -0,0 +1,63 @@
+_BASE_: Base-bagtricks.yml
+
+MODEL:
+  FREEZE_LAYERS: [ backbone ]
+
+  BACKBONE:
+    WITH_NL: True
+
+  HEADS:
+    NECK_FEAT: after
+    POOL_LAYER: GeneralizedMeanPoolingP
+    CLS_LAYER: CircleSoftmax
+    SCALE: 64
+    MARGIN: 0.35
+
+  LOSSES:
+    NAME: ("CrossEntropyLoss", "TripletLoss",)
+    CE:
+      EPSILON: 0.1
+      SCALE: 1.0
+
+    TRI:
+      MARGIN: 0.0
+      HARD_MINING: True
+      NORM_FEAT: False
+      SCALE: 1.0
+
+INPUT:
+  SIZE_TRAIN: [ 384, 128 ]
+  SIZE_TEST: [ 384, 128 ]
+
+  AUTOAUG:
+    ENABLED: True
+    PROB: 0.1
+
+DATALOADER:
+  NUM_INSTANCE: 16
+
+SOLVER:
+  AMP:
+    ENABLED: True
+  OPT: Adam
+  MAX_EPOCH: 60
+  BASE_LR: 0.00035
+  WEIGHT_DECAY: 0.0005
+  IMS_PER_BATCH: 64
+
+  SCHED: CosineAnnealingLR
+  DELAY_EPOCHS: 30
+  ETA_MIN_LR: 0.0000007
+
+  WARMUP_FACTOR: 0.1
+  WARMUP_ITERS: 2000
+
+  FREEZE_ITERS: 1000
+
+  CHECKPOINT_PERIOD: 20
+
+TEST:
+  EVAL_PERIOD: 10
+  IMS_PER_BATCH: 128
+
+CUDNN_BENCHMARK: True

dnt/third_party/fast-reid/configs/Base-bagtricks.yml

@@ -0,0 +1,76 @@
+MODEL:
+  META_ARCHITECTURE: Baseline
+
+  BACKBONE:
+    NAME: build_resnet_backbone
+    NORM: BN
+    DEPTH: 50x
+    LAST_STRIDE: 1
+    FEAT_DIM: 2048
+    WITH_IBN: False
+    PRETRAIN: True
+
+  HEADS:
+    NAME: EmbeddingHead
+    NORM: BN
+    WITH_BNNECK: True
+    POOL_LAYER: GlobalAvgPool
+    NECK_FEAT: before
+    CLS_LAYER: Linear
+
+  LOSSES:
+    NAME: ("CrossEntropyLoss", "TripletLoss",)
+
+    CE:
+      EPSILON: 0.1
+      SCALE: 1.
+
+    TRI:
+      MARGIN: 0.3
+      HARD_MINING: True
+      NORM_FEAT: False
+      SCALE: 1.
+
+INPUT:
+  SIZE_TRAIN: [ 256, 128 ]
+  SIZE_TEST: [ 256, 128 ]
+
+  REA:
+    ENABLED: True
+    PROB: 0.5
+
+  FLIP:
+    ENABLED: True
+
+  PADDING:
+    ENABLED: True
+
+DATALOADER:
+  SAMPLER_TRAIN: NaiveIdentitySampler
+  NUM_INSTANCE: 4
+  NUM_WORKERS: 8
+
+SOLVER:
+  AMP:
+    ENABLED: True
+  OPT: Adam
+  MAX_EPOCH: 120
+  BASE_LR: 0.00035
+  WEIGHT_DECAY: 0.0005
+  WEIGHT_DECAY_NORM: 0.0005
+  IMS_PER_BATCH: 64
+
+  SCHED: MultiStepLR
+  STEPS: [ 40, 90 ]
+  GAMMA: 0.1
+
+  WARMUP_FACTOR: 0.1
+  WARMUP_ITERS: 2000
+
+  CHECKPOINT_PERIOD: 30
+
+TEST:
+  EVAL_PERIOD: 30
+  IMS_PER_BATCH: 128
+
+CUDNN_BENCHMARK: True

dnt/third_party/fast-reid/configs/DukeMTMC/AGW_R101-ibn.yml

@@ -0,0 +1,12 @@
+_BASE_: ../Base-AGW.yml
+
+MODEL:
+  BACKBONE:
+    DEPTH: 101x
+    WITH_IBN: True
+
+DATASETS:
+  NAMES: ("DukeMTMC",)
+  TESTS: ("DukeMTMC",)
+
+OUTPUT_DIR: logs/dukemtmc/agw_R101-ibn

dnt/third_party/fast-reid/configs/DukeMTMC/AGW_R50-ibn.yml

@@ -0,0 +1,11 @@
+_BASE_: ../Base-AGW.yml
+
+MODEL:
+  BACKBONE:
+    WITH_IBN: True
+
+DATASETS:
+  NAMES: ("DukeMTMC",)
+  TESTS: ("DukeMTMC",)
+
+OUTPUT_DIR: logs/dukemtmc/agw_R50-ibn

dnt/third_party/fast-reid/configs/DukeMTMC/AGW_R50.yml

@@ -0,0 +1,7 @@
+_BASE_: ../Base-AGW.yml
+
+DATASETS:
+  NAMES: ("DukeMTMC",)
+  TESTS: ("DukeMTMC",)
+
+OUTPUT_DIR: logs/dukemtmc/agw_R50

dnt/third_party/fast-reid/configs/DukeMTMC/AGW_S50.yml

@@ -0,0 +1,11 @@
+_BASE_: ../Base-AGW.yml
+
+MODEL:
+  BACKBONE:
+    NAME: build_resnest_backbone
+
+DATASETS:
+  NAMES: ("DukeMTMC",)
+  TESTS: ("DukeMTMC",)
+
+OUTPUT_DIR: logs/dukemtmc/agw_S50

dnt/third_party/fast-reid/configs/DukeMTMC/bagtricks_R101-ibn.yml

@@ -0,0 +1,12 @@
+_BASE_: ../Base-bagtricks.yml
+
+MODEL:
+  BACKBONE:
+    DEPTH: 101x
+    WITH_IBN: True
+
+DATASETS:
+  NAMES: ("DukeMTMC",)
+  TESTS: ("DukeMTMC",)
+
+OUTPUT_DIR: logs/dukemtmc/bagtricks_R101-ibn

dnt/third_party/fast-reid/configs/DukeMTMC/bagtricks_R50-ibn.yml

@@ -0,0 +1,11 @@
+_BASE_: ../Base-bagtricks.yml
+
+MODEL:
+  BACKBONE:
+    WITH_IBN: True
+
+DATASETS:
+  NAMES: ("DukeMTMC",)
+  TESTS: ("DukeMTMC",)
+
+OUTPUT_DIR: logs/dukemtmc/bagtricks_R50-ibn

dnt/third_party/fast-reid/configs/DukeMTMC/bagtricks_R50.yml

@@ -0,0 +1,7 @@
+_BASE_: ../Base-bagtricks.yml
+
+DATASETS:
+  NAMES: ("DukeMTMC",)
+  TESTS: ("DukeMTMC",)
+
+OUTPUT_DIR: logs/dukemtmc/bagtricks_R50

dnt/third_party/fast-reid/configs/DukeMTMC/bagtricks_S50.yml

@@ -0,0 +1,11 @@
+_BASE_: ../Base-bagtricks.yml
+
+MODEL:
+  BACKBONE:
+    NAME: build_resnest_backbone
+
+DATASETS:
+  NAMES: ("DukeMTMC",)
+  TESTS: ("DukeMTMC",)
+
+OUTPUT_DIR: logs/dukemtmc/bagtricks_S50

dnt/third_party/fast-reid/configs/DukeMTMC/mgn_R50-ibn.yml

@@ -0,0 +1,11 @@
+_BASE_: ../Base-MGN.yml
+
+MODEL:
+  BACKBONE:
+    WITH_IBN: True
+
+DATASETS:
+  NAMES: ("DukeMTMC",)
+  TESTS: ("DukeMTMC",)
+
+OUTPUT_DIR: logs/dukemtmc/mgn_R50-ibn

dnt/third_party/fast-reid/configs/DukeMTMC/sbs_R101-ibn.yml

@@ -0,0 +1,12 @@
+_BASE_: ../Base-SBS.yml
+
+MODEL:
+  BACKBONE:
+    DEPTH: 101x
+    WITH_IBN: True
+
+DATASETS:
+  NAMES: ("DukeMTMC",)
+  TESTS: ("DukeMTMC",)
+
+OUTPUT_DIR: logs/dukemtmc/sbs_R101-ibn

dnt/third_party/fast-reid/configs/DukeMTMC/sbs_R50-ibn.yml

@@ -0,0 +1,11 @@
+_BASE_: ../Base-SBS.yml
+
+MODEL:
+  BACKBONE:
+    WITH_IBN: True
+
+DATASETS:
+  NAMES: ("DukeMTMC",)
+  TESTS: ("DukeMTMC",)
+
+OUTPUT_DIR: logs/dukemtmc/sbs_R50-ibn

dnt/third_party/fast-reid/configs/DukeMTMC/sbs_R50.yml

@@ -0,0 +1,7 @@
+_BASE_: ../Base-SBS.yml
+
+DATASETS:
+  NAMES: ("DukeMTMC",)
+  TESTS: ("DukeMTMC",)
+
+OUTPUT_DIR: logs/dukemtmc/sbs_R50

dnt/third_party/fast-reid/configs/DukeMTMC/sbs_S50.yml

@@ -0,0 +1,11 @@
+_BASE_: ../Base-SBS.yml
+
+MODEL:
+  BACKBONE:
+    NAME: build_resnest_backbone
+
+DATASETS:
+  NAMES: ("DukeMTMC",)
+  TESTS: ("DukeMTMC",)
+
+OUTPUT_DIR: logs/dukemtmc/sbs_S50

dnt/third_party/fast-reid/configs/MOT17/AGW_R101-ibn.yml

@@ -0,0 +1,12 @@
+_BASE_: ../Base-AGW.yml
+
+MODEL:
+  BACKBONE:
+    DEPTH: 101x
+    WITH_IBN: True
+
+DATASETS:
+  NAMES: ("MOT17",)
+  TESTS: ("MOT17",)
+
+OUTPUT_DIR: logs/mot17/agw_R101-ibn

dnt/third_party/fast-reid/configs/MOT17/AGW_R50-ibn.yml

@@ -0,0 +1,11 @@
+_BASE_: ../Base-AGW.yml
+
+MODEL:
+  BACKBONE:
+    WITH_IBN: True
+
+DATASETS:
+  NAMES: ("MOT17",)
+  TESTS: ("MOT17",)
+
+OUTPUT_DIR: logs/mot17/agw_R50-ibn

dnt/third_party/fast-reid/configs/MOT17/AGW_R50.yml

@@ -0,0 +1,7 @@
+_BASE_: ../Base-AGW.yml
+
+DATASETS:
+  NAMES: ("MOT17",)
+  TESTS: ("MOT17",)
+
+OUTPUT_DIR: logs/mot17/agw_R50

dnt/third_party/fast-reid/configs/MOT17/AGW_S50.yml

@@ -0,0 +1,11 @@
+_BASE_: ../Base-AGW.yml
+
+MODEL:
+  BACKBONE:
+    NAME: build_resnest_backbone
+
+DATASETS:
+  NAMES: ("MOT17",)
+  TESTS: ("MOT17",)
+
+OUTPUT_DIR: logs/mot17/agw_S50

dnt/third_party/fast-reid/configs/MOT17/bagtricks_R101-ibn.yml

@@ -0,0 +1,12 @@
+_BASE_: ../Base-bagtricks.yml
+
+MODEL:
+  BACKBONE:
+    DEPTH: 101x
+    WITH_IBN: True
+
+DATASETS:
+  NAMES: ("MOT17",)
+  TESTS: ("MOT17",)
+
+OUTPUT_DIR: logs/mot17/bagtricks_R101-ibn

dnt/third_party/fast-reid/configs/MOT17/bagtricks_R50-ibn.yml

@@ -0,0 +1,11 @@
+_BASE_: ../Base-bagtricks.yml
+
+MODEL:
+  BACKBONE:
+    WITH_IBN: True
+
+DATASETS:
+  NAMES: ("MOT17",)
+  TESTS: ("MOT17",)
+
+OUTPUT_DIR: logs/mot17/bagtricks_R50-ibn

dnt/third_party/fast-reid/configs/MOT17/bagtricks_R50.yml

@@ -0,0 +1,7 @@
+_BASE_: ../Base-bagtricks.yml
+
+DATASETS:
+  NAMES: ("MOT17",)
+  TESTS: ("MOT17",)
+
+OUTPUT_DIR: logs/mot17/bagtricks_R50

dnt/third_party/fast-reid/configs/MOT17/bagtricks_S50.yml

@@ -0,0 +1,11 @@
+_BASE_: ../Base-bagtricks.yml
+
+MODEL:
+  BACKBONE:
+    NAME: build_resnest_backbone
+
+DATASETS:
+  NAMES: ("MOT17",)
+  TESTS: ("MOT17",)
+
+OUTPUT_DIR: logs/mot17/bagtricks_S50

dnt/third_party/fast-reid/configs/MOT17/mgn_R50-ibn.yml

@@ -0,0 +1,11 @@
+_BASE_: ../Base-MGN.yml
+
+MODEL:
+  BACKBONE:
+    WITH_IBN: True
+
+DATASETS:
+  NAMES: ("MOT17",)
+  TESTS: ("MOT17",)
+
+OUTPUT_DIR: logs/mot17/mgn_R50-ibn

dnt/third_party/fast-reid/configs/MOT17/sbs_R101-ibn.yml

@@ -0,0 +1,12 @@
+_BASE_: ../Base-SBS.yml
+
+MODEL:
+  BACKBONE:
+    DEPTH: 101x
+    WITH_IBN: True
+
+DATASETS:
+  NAMES: ("MOT17",)
+  TESTS: ("MOT17",)
+
+OUTPUT_DIR: logs/mot17/sbs_R101-ibn

dnt/third_party/fast-reid/configs/MOT17/sbs_R50-ibn.yml

@@ -0,0 +1,11 @@
+_BASE_: ../Base-SBS.yml
+
+MODEL:
+  BACKBONE:
+    WITH_IBN: True
+
+DATASETS:
+  NAMES: ("MOT17",)
+  TESTS: ("MOT17",)
+
+OUTPUT_DIR: logs/mot17/sbs_R50-ibn

dnt/third_party/fast-reid/configs/MOT17/sbs_R50.yml

@@ -0,0 +1,7 @@
+_BASE_: ../Base-SBS.yml
+
+DATASETS:
+  NAMES: ("MOT17",)
+  TESTS: ("MOT17",)
+
+OUTPUT_DIR: logs/mot17/sbs_R50

dnt/third_party/fast-reid/configs/MOT17/sbs_S50.yml

@@ -0,0 +1,11 @@
+_BASE_: ../Base-SBS.yml
+
+MODEL:
+  BACKBONE:
+    NAME: build_resnest_backbone
+
+DATASETS:
+  NAMES: ("MOT17",)
+  TESTS: ("MOT17",)
+
+OUTPUT_DIR: logs/MOT17/sbs_S50

dnt/third_party/fast-reid/configs/MOT20/AGW_R101-ibn.yml

@@ -0,0 +1,12 @@
+_BASE_: ../Base-AGW.yml
+
+MODEL:
+  BACKBONE:
+    DEPTH: 101x
+    WITH_IBN: True
+
+DATASETS:
+  NAMES: ("MOT20",)
+  TESTS: ("MOT20",)
+
+OUTPUT_DIR: logs/mot20/agw_R101-ibn

dnt/third_party/fast-reid/configs/MOT20/AGW_R50-ibn.yml

@@ -0,0 +1,11 @@
+_BASE_: ../Base-AGW.yml
+
+MODEL:
+  BACKBONE:
+    WITH_IBN: True
+
+DATASETS:
+  NAMES: ("MOT20",)
+  TESTS: ("MOT20",)
+
+OUTPUT_DIR: logs/mot20/agw_R50-ibn

dnt/third_party/fast-reid/configs/MOT20/AGW_R50.yml

@@ -0,0 +1,7 @@
+_BASE_: ../Base-AGW.yml
+
+DATASETS:
+  NAMES: ("MOT20",)
+  TESTS: ("MOT20",)
+
+OUTPUT_DIR: logs/mot20/agw_R50

dnt/third_party/fast-reid/configs/MOT20/AGW_S50.yml

@@ -0,0 +1,11 @@
+_BASE_: ../Base-AGW.yml
+
+MODEL:
+  BACKBONE:
+    NAME: build_resnest_backbone
+
+DATASETS:
+  NAMES: ("MOT20",)
+  TESTS: ("MOT20",)
+
+OUTPUT_DIR: logs/mot20/agw_S50

dnt/third_party/fast-reid/configs/MOT20/bagtricks_R101-ibn.yml

@@ -0,0 +1,12 @@
+_BASE_: ../Base-bagtricks.yml
+
+MODEL:
+  BACKBONE:
+    DEPTH: 101x
+    WITH_IBN: True
+
+DATASETS:
+  NAMES: ("MOT20",)
+  TESTS: ("MOT20",)
+
+OUTPUT_DIR: logs/mot20/bagtricks_R101-ibn

dnt/third_party/fast-reid/configs/MOT20/bagtricks_R50-ibn.yml

@@ -0,0 +1,11 @@
+_BASE_: ../Base-bagtricks.yml
+
+MODEL:
+  BACKBONE:
+    WITH_IBN: True
+
+DATASETS:
+  NAMES: ("MOT20",)
+  TESTS: ("MOT20",)
+
+OUTPUT_DIR: logs/mot20/bagtricks_R50-ibn

dnt/third_party/fast-reid/configs/MOT20/bagtricks_R50.yml

@@ -0,0 +1,7 @@
+_BASE_: ../Base-bagtricks.yml
+
+DATASETS:
+  NAMES: ("MOT20",)
+  TESTS: ("MOT20",)
+
+OUTPUT_DIR: logs/mot20/bagtricks_R50

dnt/third_party/fast-reid/configs/MOT20/bagtricks_S50.yml

@@ -0,0 +1,11 @@
+_BASE_: ../Base-bagtricks.yml
+
+MODEL:
+  BACKBONE:
+    NAME: build_resnest_backbone
+
+DATASETS:
+  NAMES: ("MOT20",)
+  TESTS: ("MOT20",)
+
+OUTPUT_DIR: logs/mot20/bagtricks_S50

dnt/third_party/fast-reid/configs/MOT20/mgn_R50-ibn.yml

@@ -0,0 +1,11 @@
+_BASE_: ../Base-MGN.yml
+
+MODEL:
+  BACKBONE:
+    WITH_IBN: True
+
+DATASETS:
+  NAMES: ("MOT20",)
+  TESTS: ("MOT20",)
+
+OUTPUT_DIR: logs/mot20/mgn_R50-ibn

dnt/third_party/fast-reid/configs/MOT20/sbs_R101-ibn.yml

@@ -0,0 +1,12 @@
+_BASE_: ../Base-SBS.yml
+
+MODEL:
+  BACKBONE:
+    DEPTH: 101x
+    WITH_IBN: True
+
+DATASETS:
+  NAMES: ("MOT20",)
+  TESTS: ("MOT20",)
+
+OUTPUT_DIR: logs/mot20/sbs_R101-ibn

dnt/third_party/fast-reid/configs/MOT20/sbs_R50-ibn.yml

@@ -0,0 +1,11 @@
+_BASE_: ../Base-SBS.yml
+
+MODEL:
+  BACKBONE:
+    WITH_IBN: True
+
+DATASETS:
+  NAMES: ("MOT20",)
+  TESTS: ("MOT20",)
+
+OUTPUT_DIR: logs/mot20/sbs_R50-ibn

dnt/third_party/fast-reid/configs/MOT20/sbs_R50.yml

@@ -0,0 +1,7 @@
+_BASE_: ../Base-SBS.yml
+
+DATASETS:
+  NAMES: ("MOT20",)
+  TESTS: ("MOT20",)
+
+OUTPUT_DIR: logs/mot20/sbs_R50