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

[Bug Fix ]Update ppseg visualize python api && demo (#454)

Browse Source 
Update ppseg visualize python api && demo
This commit is contained in:
huangjianhui
2022-10-28 20:38:05 +08:00
committed by GitHub
parent 52423b897e
commit d794810f10
3 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
examples/vision/segmentation/paddleseg/cpp/infer.cc
View File

@@ -43,7 +43,7 @@ void CpuInfer(const std::string& model_dir, const std::string& image_file) {
} }
std::cout << res.Str() << std::endl; std::cout << res.Str() << std::endl;
auto vis_im = fastdeploy::vision::VisSegmentation(im, res); auto vis_im = fastdeploy::vision::VisSegmentation(im, res, 0.5);
cv::imwrite("vis_result.jpg", vis_im); cv::imwrite("vis_result.jpg", vis_im);
std::cout << "Visualized result saved in ./vis_result.jpg" << std::endl; std::cout << "Visualized result saved in ./vis_result.jpg" << std::endl;
} }
@@ -72,7 +72,7 @@ void GpuInfer(const std::string& model_dir, const std::string& image_file) {
} }
std::cout << res.Str() << std::endl; std::cout << res.Str() << std::endl;
auto vis_im = fastdeploy::vision::VisSegmentation(im, res); auto vis_im = fastdeploy::vision::VisSegmentation(im, res, 0.5);
cv::imwrite("vis_result.jpg", vis_im); cv::imwrite("vis_result.jpg", vis_im);
std::cout << "Visualized result saved in ./vis_result.jpg" << std::endl; std::cout << "Visualized result saved in ./vis_result.jpg" << std::endl;
} }
@@ -102,7 +102,7 @@ void TrtInfer(const std::string& model_dir, const std::string& image_file) {
} }
std::cout << res.Str() << std::endl; std::cout << res.Str() << std::endl;
auto vis_im = fastdeploy::vision::VisSegmentation(im, res); auto vis_im = fastdeploy::vision::VisSegmentation(im, res, 0.5);
cv::imwrite("vis_result.jpg", vis_im); cv::imwrite("vis_result.jpg", vis_im);
std::cout << "Visualized result saved in ./vis_result.jpg" << std::endl; std::cout << "Visualized result saved in ./vis_result.jpg" << std::endl;
} }

2
examples/vision/segmentation/paddleseg/python/infer.py
View File

@@ -53,5 +53,5 @@ result = model.predict(im.copy())
print(result) print(result)
# 可视化结果 # 可视化结果
vis_im = fd.vision.visualize.vis_segmentation(im, result) vis_im = fd.vision.vis_segmentation(im, result, weight=0.5)
cv2.imwrite("vis_img.png", vis_im) cv2.imwrite("vis_img.png", vis_im)

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

@@ -36,8 +36,8 @@ def vis_face_detection(im_data, face_det_result, line_size=1, font_size=0.5):
font_size) font_size)
def vis_segmentation(im_data, seg_result): def vis_segmentation(im_data, seg_result, weight=0.5):
return C.vision.vis_segmentation(im_data, seg_result) return C.vision.vis_segmentation(im_data, seg_result, weight)
def vis_matting_alpha(im_data, def vis_matting_alpha(im_data,