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: Training, don't error on loading extracted faces without stored_centering

Browse Source
This commit is contained in:
torzdf 2021-05-16 16:11:02 +01:00
parent 40798fd27f
commit 4f48e1c25b
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
lib/align/detected_face.py
View File

@ -709,7 +709,8 @@ class Mask():
``affine_matrix``, ``interpolator``, ``stored_size``, ``stored_centering``
"""
for key in ("mask", "affine_matrix", "interpolator", "stored_size", "stored_centering"):
val = mask_dict[key]
val = mask_dict.get(key)
val = "face" if key == "stored_centering" and val is None else val
if key == "affine_matrix" and not isinstance(val, np.ndarray):
val = np.array(val, dtype="float64")
setattr(self, self._attr_name(key), val)