Spaces:
Running
on
Zero
Running
on
Zero
small fix
Browse files
app.py
CHANGED
|
@@ -102,7 +102,10 @@ def process_video(
|
|
| 102 |
image=frame,
|
| 103 |
device=DEVICE
|
| 104 |
)
|
|
|
|
| 105 |
detections.confidence = np.ones(len(detections))
|
|
|
|
|
|
|
| 106 |
detections = TRACKER.update_with_detections(detections)
|
| 107 |
frame = annotate_image(
|
| 108 |
input_image=frame,
|
|
|
|
| 102 |
image=frame,
|
| 103 |
device=DEVICE
|
| 104 |
)
|
| 105 |
+
|
| 106 |
detections.confidence = np.ones(len(detections))
|
| 107 |
+
detections.class_ids = np.zeros(len(detections))
|
| 108 |
+
|
| 109 |
detections = TRACKER.update_with_detections(detections)
|
| 110 |
frame = annotate_image(
|
| 111 |
input_image=frame,
|