github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-08-24 15:16:21 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update InteractiveMergerSubprocessor.py

Browse source
This commit is contained in:
Felipe Ducau 2020-04-15 22:25:14 -03:00 • committed by GitHub
commit d63a682204
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

7
merger/InteractiveMergerSubprocessor.py
View file

@ -81,6 +81,7 @@ class InteractiveMergerSubprocessor(Subprocessor):
cfg = pf.cfg.copy() cfg = pf.cfg.copy()
frame_info = pf.frame_info frame_info = pf.frame_info
parallel_frame_info = pf.parallel_frame_info
filepath = frame_info.filepath filepath = frame_info.filepath
if len(frame_info.landmarks_list) == 0: if len(frame_info.landmarks_list) == 0:
@ -109,7 +110,9 @@ class InteractiveMergerSubprocessor(Subprocessor):
face_enhancer_func=self.face_enhancer_func, face_enhancer_func=self.face_enhancer_func,
xseg_256_extract_func=self.xseg_256_extract_func, xseg_256_extract_func=self.xseg_256_extract_func,
cfg=cfg, cfg=cfg,
frame_info=frame_info) frame_info=frame_info,
parallel_frame_info=parallel_frame_info,
)
except Exception as e: except Exception as e:
e_str = traceback.format_exc() e_str = traceback.format_exc()
if 'MemoryError' in e_str: if 'MemoryError' in e_str:
@ -570,4 +573,4 @@ class InteractiveMergerSubprocessor(Subprocessor):
#override #override
def get_result(self): def get_result(self):
return 0 return 0