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

Add new model PaddleSeg (#30)

Browse Source 
* Support new model PaddleSeg

* Fix conflict

* PaddleSeg add visulization function

* fix bug

* Fix BindPPSeg wrong name

* Fix variable name

* Update by comments

* Add ppseg-unet example python version

Co-authored-by: Jason <jiangjiajun@baidu.com>
This commit is contained in:
huangjianhui
2022-07-21 15:38:21 +08:00
committed by GitHub
parent 8b0a0c6a10
commit a8458e6729
15 changed files with 453 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -18,11 +18,20 @@ namespace fastdeploy {
void BindVisualize(pybind11::module& m) {
pybind11::class_<vision::Visualize>(m, "Visualize")
.def(pybind11::init<>())
.def_static("vis_detection", [](pybind11::array& im_data,
vision::DetectionResult& result,
int line_size, float font_size) {
auto im = PyArrayToCvMat(im_data);
vision::Visualize::VisDetection(&im, result, line_size, font_size);
.def_static("vis_detection",
[](pybind11::array& im_data, vision::DetectionResult& result,
int line_size, float font_size) {
auto im = PyArrayToCvMat(im_data);
vision::Visualize::VisDetection(&im, result, line_size,
font_size);
})
.def_static("vis_segmentation", [](pybind11::array& im_data,
vision::SegmentationResult& result,
pybind11::array& vis_im_data,
const int& num_classes) {
cv::Mat im = PyArrayToCvMat(im_data);
cv::Mat vis_im = PyArrayToCvMat(vis_im_data);
vision::Visualize::VisSegmentation(im, result, &vis_im, num_classes);
});
}
} // namespace fastdeploy
} // namespace fastdeploy