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()
|