Spaces:
Running
on
T4
Running
on
T4
PKUWilliamYang
commited on
Commit
·
a6c71e9
1
Parent(s):
aa92081
Update vtoonify_model.py
Browse files- vtoonify_model.py +1 -2
vtoonify_model.py
CHANGED
@@ -18,7 +18,6 @@ from model.encoder.align_all_parallel import align_face
|
|
18 |
import gc
|
19 |
import huggingface_hub
|
20 |
import os
|
21 |
-
from tqdm import tqdm
|
22 |
|
23 |
MODEL_REPO = 'PKUWilliamYang/VToonify'
|
24 |
|
@@ -162,7 +161,7 @@ class Model():
|
|
162 |
fourcc = cv2.VideoWriter_fourcc(*'mp4v')
|
163 |
videoWriter = cv2.VideoWriter('input.mp4', fourcc, video_cap.get(5), (int(right-left), int(bottom-top)))
|
164 |
kernel_1d = np.array([[0.125],[0.375],[0.375],[0.125]])
|
165 |
-
for i in
|
166 |
success, frame = video_cap.read()
|
167 |
frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
|
168 |
if scale <= 0.75:
|
|
|
18 |
import gc
|
19 |
import huggingface_hub
|
20 |
import os
|
|
|
21 |
|
22 |
MODEL_REPO = 'PKUWilliamYang/VToonify'
|
23 |
|
|
|
161 |
fourcc = cv2.VideoWriter_fourcc(*'mp4v')
|
162 |
videoWriter = cv2.VideoWriter('input.mp4', fourcc, video_cap.get(5), (int(right-left), int(bottom-top)))
|
163 |
kernel_1d = np.array([[0.125],[0.375],[0.375],[0.125]])
|
164 |
+
for i in range(num-1):
|
165 |
success, frame = video_cap.read()
|
166 |
frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
|
167 |
if scale <= 0.75:
|