YANGYYYY commited on
Commit
d47d493
1 Parent(s): e7d45cf

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -258,7 +258,7 @@ class Predictor:
258
  frame_count = len(video_frames)
259
  transformed_video_frames = []
260
 
261
- batch_shape = (batch_size,) + video_frames[0].shape
262
  frames = np.zeros(batch_shape, dtype=np.uint8)
263
  frame_idx = 0
264
 
 
258
  frame_count = len(video_frames)
259
  transformed_video_frames = []
260
 
261
+ batch_shape = (batch_size) + video_frames[0].shape
262
  frames = np.zeros(batch_shape, dtype=np.uint8)
263
  frame_idx = 0
264