PKUWilliamYang commited on
Commit
a6c71e9
1 Parent(s): aa92081

Update vtoonify_model.py

Browse files
Files changed (1) hide show
  1. 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 tqdm(range(num-1)):
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: