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: CV2 error on previews

Browse Source
This commit is contained in:
torzdf 2022-08-22 21:31:51 +01:00
parent 66845ea5f0
commit 0837273d46
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -665,7 +665,8 @@ class _Samples(): # pylint:disable=too-few-public-methods
"""
scale = target_size / sample.shape[1]
if scale == 1.0:
return sample
# cv2 complains if we don't do this :/
return np.ascontiguousarray(sample)
logger.debug("Resizing sample: (side: '%s', sample.shape: %s, target_size: %s, scale: %s)",
side, sample.shape, target_size, scale)
interpn = cv2.INTER_CUBIC if scale > 1.0 else cv2.INTER_AREA