nodes/dchatel-comfyui_facetools
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

wrapped warpAffine with clip(x,0,1) to ensure the image is still a proper image

Browse Source
This commit is contained in:
David Chatel
2025-02-16 18:38:12 +01:00
parent dfa90ae487
commit 51380df417
2 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes.py
+4 -4
View File
@@ -170,18 +170,18 @@ class WarpFaceBack:
else:
values = groups[i]
crop, mask, warp = list(zip(*[x[1:] for x in values]))
warped_masks = [cv2.warpAffine(single_mask.numpy(),
warped_masks = [np.clip(cv2.warpAffine(single_mask.numpy(),
cv2.invertAffineTransform(single_warp),
image.shape[1::-1],
flags=cv2.INTER_LANCZOS4)
flags=cv2.INTER_LANCZOS4), 0, 1)
for single_warp, single_mask in zip(warp, mask)]
full_mask = np.add.reduce(warped_masks, axis=0)[...,None]
swapped = np.add.reduce([
cv2.warpAffine(single_crop.cpu().numpy(),
np.clip(cv2.warpAffine(single_crop.cpu().numpy(),
cv2.invertAffineTransform(single_warp),
image.shape[1::-1],
flags=cv2.INTER_LANCZOS4
) * single_mask[..., None]
), 0, 1) * single_mask[..., None]
for single_crop, single_mask, single_warp in zip(crop, warped_masks, warp)
], axis=0) / np.maximum(1, full_mask)
full_mask = np.minimum(1, full_mask)
utils.py
+1
View File
@@ -121,6 +121,7 @@ class Face:
N = N @ T4 @ S @ T3
crop = cv2.warpAffine(self.img.numpy(), N, (size, size), flags=cv2.INTER_LANCZOS4)
crop = torch.from_numpy(crop)[None]
crop = torch.clip(crop, 0, 1)
return N, crop