add centroid asso test

mikel-brostrom · Jan 12, 2024 · 075bec6 · 075bec6
1 parent b4cd344
commit 075bec6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_python.py b/tests/test_python.py
@@ -57,8 +57,8 @@ def test_deepocsort_output():
         device='cpu',
         half=False,
         per_class=False,
-        asso_func="centroid"
     )
+    tracker.asso_func="centroid"
     rgb = np.random.randint(255, size=(640, 640, 3), dtype=np.uint8)
     det = np.array([[144, 212, 578, 480, 0.82, 0],
                     [425, 281, 576, 472, 0.56, 65]])