Upload 4 files

.gitattributes

@@ -32,3 +32,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+01.mp4 filter=lfs diff=lfs merge=lfs -text

01.mp4

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7a540354befcb25f1df6fa4a823884092407f52e4f484f917bc98c895549fe69
+size 1276742

app.py

@@ -0,0 +1,90 @@
+import os
+os.system("pip3 install cython_bbox gdown 'git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI'")
+from torchyolo import YoloHub
+import gradio as gr
+from utils import attempt_download_from_hub
+
+
+
+def object_tracker(
+    source: str,
+    model_type: str,
+    model_path: str,
+    tracker_type: str,
+    tracker_config_path: str,
+    StrongSort_OsNet_Path: str = None,
+):
+    model = YoloHub(
+        config_path="default_config.yaml",
+        model_type=model_type,
+        model_path=model_path,
+    )
+
+    if tracker_type == "STRONGSORT":
+        StrongSort_OsNet_Path = attempt_download_from_hub(StrongSort_OsNet_Path)
+
+    model.predict(
+        source=source,
+        tracker_type=tracker_type,
+        tracker_weight_path=StrongSort_OsNet_Path,
+        tracker_config_path=tracker_config_path,
+    )
+
+    return 'output.mp4'
+       
+
+inputs = [
+    gr.Video(),
+    gr.inputs.Dropdown(
+        label="Model Type",
+        choices=["yolov8"],
+        default="yolov8",
+    ),
+    gr.inputs.Dropdown(
+        label="Model Path",
+        choices=[
+            "aijack/yv8n"        ],
+        default="aijack/yv8n",
+    ),
+    gr.inputs.Dropdown(
+        label="Tracker Type",
+        choices=["OCSORT"],
+        default="OCSORT",
+    ),
+    gr.inputs.Dropdown(
+        label="Tracker Config Path",
+        choices=[
+            "tracker/oc_sort.yaml",
+        ],
+        default="tracker/oc_sort.yaml",
+    ),
+    gr.inputs.Dropdown(
+        label="Tracker Weight Path",
+        choices=[
+            "aijack/osnet"
+        ],
+        default="aijack/osnet",
+    ),
+]
+examples = [
+    [
+        "01.mp4",
+        "yolov8",
+        "aijack/yv8n",
+        "OCSORT",
+        "tracker/oc_sort.yaml",
+    ]
+]
+outputs = gr.Video()
+title = "YOLO8 Object Detection and Track Algorithm Library"
+article = "<p style='text-align: center'><a href='http://claireye.com.tw'>Claireye</a> | 2023</p>"
+demo_app = gr.Interface(
+    fn=object_tracker,
+    inputs=inputs,
+    examples=examples,
+    outputs=outputs,
+    title=title,
+	article = article,
+    cache_examples=False
+)
+demo_app.launch(debug=True, enable_queue=True)

requirements.txt

@@ -0,0 +1,8 @@
+Cython
+ultralytics
+torchyolo==1.1.1
+norfair-tracker
+ocsort
+sort-track
+bytetracker==0.3.1
+strongsort==0.2.1

utils.py

@@ -0,0 +1,16 @@
+def attempt_download_from_hub(repo_id, hf_token=None):
+    from huggingface_hub import hf_hub_download, list_repo_files
+    from huggingface_hub.utils._errors import RepositoryNotFoundError
+    from huggingface_hub.utils._validators import HFValidationError
+    try:
+        repo_files = list_repo_files(repo_id=repo_id, repo_type='model', token=hf_token)
+        model_file = [f for f in repo_files if f.endswith('.pt')][0]
+        file = hf_hub_download(
+            repo_id=repo_id,
+            filename=model_file,
+            repo_type='model',
+            token=hf_token,
+        )
+        return file
+    except (RepositoryNotFoundError, HFValidationError):
+        return None