PKUWilliamYang commited on
Commit
e059e75
1 Parent(s): 426dd72

Update vtoonify_model.py

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