Spaces:

k20hcmus
/

FishEye8K

Running on Zero

tuansunday05 commited on May 12, 2024

Commit

07c482c

verified ·

1 Parent(s): 846ed57

Update detect_deepsort.py

Files changed (1) hide show

detect_deepsort.py CHANGED Viewed

@@ -200,7 +200,7 @@ def run_deepsort(
         with dt[1]:
             visualize = increment_path(save_dir / Path(path).stem, mkdir=True) if visualize else False
             pred = model(im, augment=augment, visualize=visualize)
-            pred = pred[0][1]
         # NMS
         with dt[2]:

         with dt[1]:
             visualize = increment_path(save_dir / Path(path).stem, mkdir=True) if visualize else False
             pred = model(im, augment=augment, visualize=visualize)
+            # pred = pred[0][1]
         # NMS
         with dt[2]: