jones/facefusion
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Polish paste back

Browse Source
This commit is contained in:
henryruhs
2025-06-16 22:46:17 +02:00
parent 9235802b03
commit a8655f58b6
1 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
facefusion/face_helper.py
View File

@@ -106,11 +106,10 @@ def paste_back(temp_vision_frame : VisionFrame, crop_vision_frame : VisionFrame,
inverse_mask = cv2.warpAffine(crop_mask, paste_matrix, (paste_width, paste_height)).clip(0, 1)
inverse_mask = numpy.expand_dims(inverse_mask, axis = -1)
inverse_vision_frame = cv2.warpAffine(crop_vision_frame, paste_matrix, (paste_width, paste_height), borderMode = cv2.BORDER_REPLICATE)
paste_vision_frame = temp_vision_frame.copy()
paste_region_frame = paste_vision_frame[y_min:y_max, x_min:x_max]
paste_region_frame = paste_region_frame * (1 - inverse_mask) + inverse_vision_frame * inverse_mask
paste_vision_frame[y_min:y_max, x_min:x_max] = paste_region_frame.astype(temp_vision_frame.dtype)
return paste_vision_frame
paste_vision_frame = temp_vision_frame[y_min:y_max, x_min:x_max]
paste_vision_frame = paste_vision_frame * (1 - inverse_mask) + inverse_vision_frame * inverse_mask
temp_vision_frame[y_min:y_max, x_min:x_max] = paste_vision_frame.astype(temp_vision_frame.dtype)
return temp_vision_frame
def calc_paste_area(temp_vision_frame : VisionFrame, crop_vision_frame : VisionFrame, affine_matrix : Matrix) -> Tuple[BoundingBox, Matrix]: