OSSForks/faceswap
1
0
Fork 0
mirror of https://github.com/zebrajr/faceswap.git synced 2025-12-06 00:20:09 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

bugfix: manual tool - alignments saving

Browse Source
This commit is contained in:
torzdf 2023-06-28 18:43:37 +01:00
parent efa4471243
commit 4b20316975
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tools/manual/detected_faces.py
View File

@ -300,7 +300,8 @@ class _DiskIO(): # pylint:disable=too-few-public-methods
logger.verbose("Saving alignments for %s updated frames", # type:ignore[attr-defined]
len(frames))
for idx, faces in zip(frames, np.array(self._frame_faces)[np.array(frames)]):
for idx, faces in zip(frames,
np.array(self._frame_faces, dtype="object")[np.array(frames)]):
frame = self._sorted_frame_names[idx]
self._alignments.data[frame]["faces"] = [face.to_alignment() for face in faces]