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

lib.training - Outpu useful error message if non-faceswap training images used

Browse Source
This commit is contained in:
torzdf 2024-03-22 12:51:50 +00:00
parent aaa63ff896
commit 180fe97a52
1 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
lib/training/cache.py
View File

@ -225,8 +225,17 @@ class _Cache():
logger.debug("All metadata already cached for: %s", keys)
return read_image_batch(filenames)
batch, metadata = read_image_batch(filenames, with_metadata=True)
try:
batch, metadata = read_image_batch(filenames, with_metadata=True)
except ValueError as err:
if "inhomogeneous" in str(err):
raise FaceswapError(
"There was an error loading a batch of images. This is most likely due to "
"non-faceswap extracted faces in your training folder."
"\nAll training images should be Faceswap extracted faces."
"\nAll training images should be the same size."
f"\nThe files that caused this error are: {filenames}") from err
raise
if len(batch.shape) == 1:
folder = os.path.dirname(filenames[0])
details = [