Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
8f3cd14
1
Parent(s):
a6436e2
Update app
Browse files
preprocess/inference_preprocess.py
CHANGED
|
@@ -166,9 +166,9 @@ def inference_video(opt, padding=0):
|
|
| 166 |
videofile = os.path.join(opt.avi_dir, 'video.avi')
|
| 167 |
vidObj = cv2.VideoCapture(videofile)
|
| 168 |
yolo_model = YOLO("yolov9t.pt")
|
|
|
|
| 169 |
dets = []
|
| 170 |
fidx = 0
|
| 171 |
-
global dets, fidx
|
| 172 |
|
| 173 |
print("Detecting people in the video using YOLO (slowest step in the pipeline)...")
|
| 174 |
def generate_detections():
|
|
|
|
| 166 |
videofile = os.path.join(opt.avi_dir, 'video.avi')
|
| 167 |
vidObj = cv2.VideoCapture(videofile)
|
| 168 |
yolo_model = YOLO("yolov9t.pt")
|
| 169 |
+
global dets, fidx
|
| 170 |
dets = []
|
| 171 |
fidx = 0
|
|
|
|
| 172 |
|
| 173 |
print("Detecting people in the video using YOLO (slowest step in the pipeline)...")
|
| 174 |
def generate_detections():
|