PKUWilliamYang commited on
Commit
eaba2a3
1 Parent(s): d376f4c

Update vtoonify_model.py

Browse files
Files changed (1) hide show
  1. vtoonify_model.py +1 -1
vtoonify_model.py CHANGED
@@ -242,7 +242,7 @@ class Model():
242
  batch_size = min(max(1, int(4 * 400 * 360/ video_cap.get(3) / video_cap.get(4))), 4)
243
  else:
244
  batch_size = 1
245
- print('*** Toonify using batch size of %d on %dx%d video of %d frames'%(batch_size, int(video_cap.get(3)*4), int(video_cap.get(4)*4, num))
246
  with torch.no_grad():
247
  if self.color_transfer:
248
  s_w = exstyle
 
242
  batch_size = min(max(1, int(4 * 400 * 360/ video_cap.get(3) / video_cap.get(4))), 4)
243
  else:
244
  batch_size = 1
245
+ print('*** Toonify using batch size of %d on %dx%d video of %d frames'%(batch_size, int(video_cap.get(3)*4), int(video_cap.get(4)*4), num))
246
  with torch.no_grad():
247
  if self.color_transfer:
248
  s_w = exstyle