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: Alignments - correctly update keys for multiple underscores in the filename

Browse Source
This commit is contained in:
torzdf 2024-04-26 17:39:46 +01:00
parent 0f947791f5
commit 13dd5b3a90
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
lib/align/updater.py
View File

@ -140,17 +140,19 @@ class VideoExtension(_Updater):
"""
updated = 0
for key in list(self._alignments.data):
val = self._alignments.data[key]
fname = os.path.splitext(key)[0]
if fname.rsplit("_")[0] != self._video_name:
if fname.rsplit("_", maxsplit=1)[0] != self._video_name:
continue # Key is from a different source
val = self._alignments.data[key]
new_key = f"{fname}{self._extension}"
del self._alignments.data[key]
self._alignments.data[new_key] = val
updated += 1
logger.debug("Updated alignemnt keys for video extension: %s", updated)
logger.debug("Updated alignment keys for video extension: %s", updated)
return updated