From 6f73fc0471560cdde977568ec6566f35e0600fad Mon Sep 17 00:00:00 2001 From: iperov Date: Wed, 17 Nov 2021 18:21:10 +0400 Subject: [PATCH] fix --- xlib/face/FaceWarper.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/xlib/face/FaceWarper.py b/xlib/face/FaceWarper.py index c60f81b..04fbbd2 100644 --- a/xlib/face/FaceWarper.py +++ b/xlib/face/FaceWarper.py @@ -88,8 +88,9 @@ class FaceWarper: if data is None: rnd_state = np.random.RandomState() rnd_state.set_state( self._rnd_state_state ) + self._cached[key] = data = self._gen(H,W, random_warp, out_res, rnd_state=rnd_state ) - image_grid, face_mask = self._cached[key] = self._gen(H,W, random_warp, out_res, rnd_state=rnd_state ) + image_grid, face_mask = data new_img = cv2.remap(img, image_grid, None, interpolation=cv2.INTER_LANCZOS4) new_img *= face_mask @@ -128,7 +129,8 @@ class FaceWarper: def _gen_random_warp_uni_grid_diff(size: int, cell_count, cell_mod, rnd_state) -> np.ndarray: """ - generates square uniform random warp + generates square uniform random warp coordinate differences + grid of shape (size, size, 2) (x,y) cell_count(3) 3+