Spaces:
Running
Running
Update inference.py
Browse files- inference.py +1 -1
inference.py
CHANGED
@@ -281,7 +281,7 @@ class Predictor:
|
|
281 |
# batch_shape = (batch_size) + video_frames[0].shape
|
282 |
# frames = np.zeros(batch_shape, dtype=np.uint8)
|
283 |
# frame_idx = 0
|
284 |
-
|
285 |
frames = np.zeros(batch_shape, dtype=np.uint8)
|
286 |
frame_idx = 0
|
287 |
|
|
|
281 |
# batch_shape = (batch_size) + video_frames[0].shape
|
282 |
# frames = np.zeros(batch_shape, dtype=np.uint8)
|
283 |
# frame_idx = 0
|
284 |
+
batch_shape = (batch_size, frame_height, frame_width, 3)
|
285 |
frames = np.zeros(batch_shape, dtype=np.uint8)
|
286 |
frame_idx = 0
|
287 |
|