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

bug fix: mssing frames

Browse source
This commit is contained in:
Jan 2021-11-26 12:07:56 +01:00
commit 3304dfcf14
1 changed files with 8 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

11
mainscripts/Merger.py
View file

@ -146,17 +146,20 @@ def main (model_class_name=None,
io.log_info ("Use 'recover original filename' to determine the exact duplicates.") io.log_info ("Use 'recover original filename' to determine the exact duplicates.")
io.log_info ("") io.log_info ("")
# build frames maunally # build frames maunally
frames = [] frames = []
for p in input_path_image_paths: for p in input_path_image_paths:
path = Path(p) cur_path = Path(p)
data = alignments.get(path.stem, None) data = alignments.get(cur_path.stem, None)
if data == None: if data == None:
frame = InteractiveMergerSubprocessor.Frame(FrameInfo(frame_info=frame_info)) frame_info=FrameInfo(filepath=cur_path)
frame = InteractiveMergerSubprocessor.Frame(frame_info=frame_info)
else: else:
landmarks_list = [d[0] for d in data] landmarks_list = [d[0] for d in data]
dfl_images_list = [d[1] for d in data] dfl_images_list = [d[1] for d in data]
frame_info=FrameInfo(filepath=path, landmarks_list=landmarks_list, dfl_images_list=dfl_images_list) frame_info=FrameInfo(filepath=cur_path, landmarks_list=landmarks_list, dfl_images_list=dfl_images_list)
frame = InteractiveMergerSubprocessor.Frame(frame_info=frame_info) frame = InteractiveMergerSubprocessor.Frame(frame_info=frame_info)
frames.append(frame) frames.append(frame)