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

[Other] Add comments for swap_background api (#941)

Browse Source 
* Update keypointdetection result docs

* Update im.copy() to im in examples

* Update new Api, fastdeploy::vision::Visualize to fastdeploy::vision

* Update SwapBackgroundSegmentation && SwapBackgroundMatting to SwapBackground

* Update README_CN.md

* Update README_CN.md

* Add comments for swap_background Api
This commit is contained in:
huangjianhui
2022-12-22 10:14:36 +08:00
committed by GitHub
parent d4faddc0aa
commit 60b84ef315
2 changed files with 29 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
python/fastdeploy/vision/visualize/__init__.py
View File

@@ -138,6 +138,15 @@ def swap_background(im_data,
result,
remove_small_connected_area=False,
background_label=0):
"""Swap the image background with MattingResult or SegmentationResult
:param im_data: (numpy.ndarray)The input image data, 3-D array with layout HWC, BGR format
:param background: (numpy.ndarray)The background image data, 3-D array with layout HWC, BGR format
:param result: The result produced by model, MattingResult or SegmentationResult
:param remove_small_connected_area: (bool) If remove_small_connected_area==True, the visualized result will not include the small connected areas
:param background_label: (int)The background label number in SegmentationResult
:return: (numpy.ndarray) image with visualized results
"""
if isinstance(result, C.vision.MattingResult):
return C.vision.swap_background(im_data, background, result,
remove_small_connected_area)