Spaces:
Sleeping
Sleeping
Update run.py
Browse files
run.py
CHANGED
@@ -43,7 +43,7 @@ def stream_object_detection(video, conf_threshold):
|
|
43 |
if n_frames % SUBSAMPLE == 0:
|
44 |
batch.append(frame)
|
45 |
if len(batch) == 2 * desired_fps:
|
46 |
-
inputs = image_processor(images=batch, return_tensors="pt")
|
47 |
|
48 |
print(f"starting batch of size {len(batch)}")
|
49 |
start = time.time()
|
|
|
43 |
if n_frames % SUBSAMPLE == 0:
|
44 |
batch.append(frame)
|
45 |
if len(batch) == 2 * desired_fps:
|
46 |
+
inputs = image_processor(images=batch, return_tensors="pt")#.to("cuda")
|
47 |
|
48 |
print(f"starting batch of size {len(batch)}")
|
49 |
start = time.time()
|