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: Alignment file video key lookup for very old alignment files

Browse Source
This commit is contained in:
torzdf 2024-04-21 19:55:45 +01:00
parent 696692dc08
commit be42b04064
1 changed files with 20 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
lib/align/updater.py
View File

@ -103,17 +103,32 @@ class VideoExtension(_Updater):
super().__init__(alignments) super().__init__(alignments)
def test(self) -> bool: def test(self) -> bool:
""" Requires update if alignments version is < 2.4 """ Requires update if the extension of the key in the alignment file is not the same
as for the input video file
Returns Returns
------- -------
bool bool
``True`` if the key extensions need updating otherwise ``False`` ``True`` if the key extensions need updating otherwise ``False``
""" """
retval = self._alignments.version < 2.4 and self._extension in VIDEO_EXTENSIONS if self._alignments.version > 2.4:
logger.debug("Needs update for video extension: %s (version: %s, extension: %s)", return False
retval, self._alignments.version, self._extension)
return retval if self._extension.lower() not in VIDEO_EXTENSIONS:
return False
exts = set(os.path.splitext(k)[-1] for k in self._alignments.data)
if len(exts) != 1:
logger.debug("Alignments file has multiple key extensions. Skipping")
return False
if self._extension in exts:
logger.debug("Alignments file contains correct key extensions. Skipping")
return False
logger.debug("Needs update for video extension (version: %s, extension: %s)",
self._alignments.version, self._extension)
return True
def update(self) -> int: def update(self) -> int:
""" Update alignments files that have been extracted from videos to have the key end in the """ Update alignments files that have been extracted from videos to have the key end in the