yolox code and almost mmdetection

UTAustin-SwarmLab · Oct 22, 2023 · 8ffb2c1 · 8ffb2c1
1 parent f7db178
commit 8ffb2c1
Show file tree

Hide file tree

Showing 7 changed files with 820 additions and 10 deletions.
diff --git a/ns_vfs/config/config.yaml b/ns_vfs/config/config.yaml
@@ -15,3 +15,10 @@ GROUNDING_DINO:
 
 YOLO:
   YOLO_CHECKPOINT_PATH:
+
+YOLOX:
+  YOLOX_CHECKPOINT_PATH:
+
+MMDETECTION:
+  MMDETECTION_CONFIG_PATH:
+  MMDETECTION_CHECKPOINT_PATH:
diff --git a/ns_vfs/config/loader.py b/ns_vfs/config/loader.py
@@ -31,4 +31,20 @@ def load_config():
         "weights",
         "yolov8n.pt",
     )
+    config.YOLOX.YOLOX_CHECKPOINT_PATH = os.path.join(
+        config.VERSION_AND_PATH.ARTIFACTS_PATH,
+        "weights",
+        "yolox_x.pth",
+    )
+    config.MMDETECTION.MMDETECTION_CONFIG_PATH = os.path.join(
+        config.VERSION_AND_PATH.ROOT_PATH,
+        "ns_vfs",
+        "config",
+        "rtmdet_tiny_8xb32-300e_coco.py",
+    )
+    config.MMDETECTION.MMDETECTION_CHECKPOINT_PATH = os.path.join(
+        config.VERSION_AND_PATH.ARTIFACTS_PATH,
+        "weights",
+        "rtmdet_tiny_8xb32-300e_coco_20220902_112414-78e30dcc.pth"
+    )
     return config