Update visualize_pybind.cc

This commit is contained in:
huangjianhui
2023-02-14 11:21:44 +08:00
committed by GitHub
parent 619b561817
commit 7bd8bbfb84

View File

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