admin/LC_NeoRefacer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fixed errors in tqdm implementation for progress tracking

Browse Source
This commit is contained in:
Xavi Vinaixa
2023-06-05 09:50:57 +02:00
parent 0a65a4032a
commit 9df3a83d7e
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
refacer.py
View File

@@ -76,8 +76,6 @@ class Refacer:
flag, frame = cap.read() flag, frame = cap.read()
if flag and len(frame)>0: if flag and len(frame)>0:
#pos_frame = int(cap.get(cv2.CAP_PROP_POS_FRAMES)) #pos_frame = int(cap.get(cv2.CAP_PROP_POS_FRAMES))
pbar.update(1)
faces = self.face_app.get(frame) faces = self.face_app.get(frame)
res = frame.copy() res = frame.copy()
@@ -88,9 +86,11 @@ class Refacer:
res = self.face_swapper.get(res, face, rep_face[1], paste_back=True) res = self.face_swapper.get(res, face, rep_face[1], paste_back=True)
output.write(res) output.write(res)
pbar.update(1)
else: else:
break break
pbar.close()
cap.release() cap.release()
output.release() output.release()