Generate a black border in the result

[superhero-7]

The result has a black border, how to adjust this?

[m4xig1]

Hi @superhero-7,
I believe this is due to an incorrect affine transformation that the authors create during the inference. If the faces are aligned, you can simply resize the PIL images as shown below:

# warp_mat_crop = datasets_faceswap.transformation_from_points(
#     bbox_pts4, datasets_faceswap.mean_box_lm4p_512
# )
# src_im_crop512 = cv2.warpAffine(
#     np.array(src_im_pil), warp_mat_crop, (512, 512), flags=cv2.INTER_LINEAR
# )

src_im_pil = src_im_pil.resize((512, 512))