apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-18 22:44:39 +08:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

Update api && comments

Browse Source
This commit is contained in:
felixhjh
2023-02-14 03:18:54 +00:00
parent 0a6ce5a6ed
commit 619b561817
4 changed files with 37 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
fastdeploy/vision/visualize/visualize_pybind.cc
View File

@@ -102,10 +102,12 @@ void BindVisualize(pybind11::module& m) {
})
.def("vis_matting",
[](pybind11::array& im_data, vision::MattingResult& result,
bool transparent_background, bool transparent_threshold,
bool remove_small_connected_area) {
cv::Mat im = PyArrayToCvMat(im_data);
auto vis_im =
vision::VisMatting(im, result, remove_small_connected_area);
auto vis_im = vision::VisMatting(
im, result, transparent_background, transparent_threshold,
remove_small_connected_area);
FDTensor out;
vision::Mat(vis_im).ShareWithTensor(&out);
return TensorToPyArray(out);