Update vtoonify_model.py

vtoonify_model.py

@@ -206,7 +206,10 @@ class Model():
                                       int(video_cap.get(4)*4)))        
 
         batch_frames = []
-        batch_size = max(1, int(round(4 * 256 * 256 / video_cap.get(3) / video_cap.get(4))))
+        if video_cap.get(3) != 0:
+            batch_size = min(max(1, int(round(4 * 400 * 360 / video_cap.get(3) / video_cap.get(4)))), 8)
+        else:
+            batch_size = 4
         print('Using batch size of %d on %d frames'%(batch_size, num))
         with torch.no_grad():
             if self.color_transfer: