Update SwapBackgroundSegmentation && SwapBackgroundMatting to SwapBackground

This commit is contained in:
felixhjh
2022-12-12 10:59:15 +00:00
parent 631593b23e
commit 37359c65f3
22 changed files with 62 additions and 113 deletions

View File

@@ -56,7 +56,7 @@ result = model.predict(im)
print(result)
# 可视化结果
vis_im = fd.vision.vis_matting_alpha(im, result)
vis_im_with_bg = fd.vision.swap_background_matting(im, bg, result)
vis_im_with_bg = fd.vision.swap_background(im, bg, result)
cv2.imwrite("visualized_result_fg.jpg", vis_im)
cv2.imwrite("visualized_result_replaced_bg.jpg", vis_im_with_bg)
print(