github/DeepFaceLive
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLive synced 2025-08-19 13:09:58 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix

Browse source
This commit is contained in:
iperov 2021-11-17 18:21:10 +04:00
commit 6f73fc0471
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
xlib/face/FaceWarper.py
View file

@ -88,8 +88,9 @@ class FaceWarper:
if data is None: if data is None:
rnd_state = np.random.RandomState() rnd_state = np.random.RandomState()
rnd_state.set_state( self._rnd_state_state ) 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 = cv2.remap(img, image_grid, None, interpolation=cv2.INTER_LANCZOS4)
new_img *= face_mask 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: 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) grid of shape (size, size, 2) (x,y)
cell_count(3) 3+ cell_count(3) 3+