Spaces:
Running
Running
Commit
·
ebee748
1
Parent(s):
a03be25
Update vtoonify_model.py
Browse files- 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(
|
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))
|