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

Added Lanczos to warpAffine

Browse Source
This commit is contained in:
David Chatel
2025-02-11 17:15:48 +01:00
parent 7176d84c5e
commit dfa90ae487
2 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes.py
+4 -2
View File
@@ -172,13 +172,15 @@ class WarpFaceBack:
crop, mask, warp = list(zip(*[x[1:] for x in values]))
warped_masks = [cv2.warpAffine(single_mask.numpy(),
cv2.invertAffineTransform(single_warp),
image.shape[1::-1])
image.shape[1::-1],
flags=cv2.INTER_LANCZOS4)
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(),
cv2.invertAffineTransform(single_warp),
image.shape[1::-1]
image.shape[1::-1],
flags=cv2.INTER_LANCZOS4
) * single_mask[..., None]
for single_crop, single_mask, single_warp in zip(crop, warped_masks, warp)
], axis=0) / np.maximum(1, full_mask)