OSSForks/faceswap
1
0
Fork 0
mirror of https://github.com/zebrajr/faceswap.git synced 2026-01-15 12:15:15 +00: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
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 updated = 0
for key in list(self._alignments.data): for key in list(self._alignments.data):
val = self._alignments.data[key]
fname = os.path.splitext(key)[0] 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 continue # Key is from a different source
val = self._alignments.data[key]
new_key = f"{fname}{self._extension}" new_key = f"{fname}{self._extension}"
del self._alignments.data[key] del self._alignments.data[key]
self._alignments.data[new_key] = val self._alignments.data[new_key] = val
updated += 1 updated += 1
logger.debug("Updated alignemnt keys for video extension: %s", updated) logger.debug("Updated alignment keys for video extension: %s", updated)
return updated return updated