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: Color order on preview with learn_mask

Browse Source
This commit is contained in:
torzdf
2023-02-14 13:09:46 +00:00
parent 7887703ac7
commit 0c1d1e3253
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
plugins/train/trainer/_base.py
View File

@@ -813,8 +813,7 @@ class _Samples(): # pylint:disable=too-few-public-methods
if self._model.color_order.lower() == "rgb": # Switch color order for RGB model display
full = full[..., ::-1]
faces = faces[..., ::-1]
predictions = [pred[..., ::-1] if pred.shape[-1] == 3 else pred
for pred in predictions]
predictions = [pred[..., 2::-1] for pred in predictions]
full = self._process_full(side, full, predictions[0].shape[1], (0., 0., 1.0))
images = [faces] + predictions