Update InteractiveMergerSubprocessor.py

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

View file

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