apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-06 09:07:10 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Update SwapBackgroundSegmentation && SwapBackgroundMatting to SwapBackground

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/vision/matting/rvm/python/infer.py
View File

@@ -86,7 +86,7 @@ if args.video is not None:
break
result = model.predict(frame)
vis_im = fd.vision.vis_matting(frame, result)
vis_im_with_bg = fd.vision.swap_background_matting(frame, bg, result)
vis_im_with_bg = fd.vision.swap_background(frame, bg, result)
alpha.write(vis_im)
composition.write(vis_im_with_bg)
cv2.waitKey(30)
@@ -104,7 +104,7 @@ if args.image is not None:
print(result)
# 可视化结果
vis_im = fd.vision.vis_matting(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(