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: preview color order for RGB models

Browse Source
This commit is contained in:
torzdf 2022-09-07 10:58:32 +01:00
parent c60aca8fb2
commit 681b775c7b
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lib/training/generator.py
View File

@ -707,7 +707,9 @@ class PreviewDataGenerator(DataGenerator):
"batch: %s, detected_faces: %s)", self._side, filenames, images.shape,
batch.shape, len(detected_faces))
self._set_color_order(batch) # Switch color order for RGB models
# Switch color order for RGB models
self._set_color_order(batch)
self._set_color_order(images)
if not self._use_mask:
mask = np.zeros_like(batch[..., 0])[..., None] + 255