PKUWilliamYang commited on
Commit
ebee748
1 Parent(s): a03be25

Update vtoonify_model.py

Browse files
Files changed (1) hide show
  1. vtoonify_model.py +1 -1
vtoonify_model.py CHANGED
@@ -207,7 +207,7 @@ class Model():
207
 
208
  batch_frames = []
209
  if video_cap.get(3) != 0:
210
- batch_size = max(1, int(round(4 * 256* 256/ video_cap.get(3) / video_cap.get(4))))
211
  else:
212
  batch_size = 1
213
  print('Using batch size of %d on %d frames'%(batch_size, num))
 
207
 
208
  batch_frames = []
209
  if video_cap.get(3) != 0:
210
+ batch_size = max(1, int(4 * 256* 256/ video_cap.get(3) / video_cap.get(4)))
211
  else:
212
  batch_size = 1
213
  print('Using batch size of %d on %d frames'%(batch_size, num))