YANGYYYY commited on
Commit
195428b
1 Parent(s): da4141c

Update inference.py

Browse files
Files changed (1) hide show
  1. 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
- _shape = (batch_size, frame_height, frame_width, 3)
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