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

Save png format image in matting example to support transparent background

Browse Source
This commit is contained in:
felixhjh
2023-02-15 07:33:45 +00:00
parent dbf446af85
commit 3835d06654
5 changed files with 11 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
examples/vision/segmentation/ppmatting/cpu-gpu/cpp/infer.cc
View File

@@ -44,7 +44,7 @@ void CpuInfer(const std::string& model_dir, const std::string& image_file,
auto vis_im = fastdeploy::vision::VisMatting(im, res);
auto vis_im_with_bg = fastdeploy::vision::SwapBackground(im, bg, res);
cv::imwrite("visualized_result.jpg", vis_im_with_bg);
cv::imwrite("visualized_result_fg.jpg", vis_im);
cv::imwrite("visualized_result_fg.png", vis_im);
std::cout << "Visualized result save in ./visualized_result_replaced_bg.jpg "
"and ./visualized_result_fg.jpg"
<< std::endl;
@@ -74,7 +74,7 @@ void KunlunXinInfer(const std::string& model_dir, const std::string& image_file,
auto vis_im = fastdeploy::vision::VisMatting(im, res);
auto vis_im_with_bg = fastdeploy::vision::SwapBackground(im, bg, res);
cv::imwrite("visualized_result.jpg", vis_im_with_bg);
cv::imwrite("visualized_result_fg.jpg", vis_im);
cv::imwrite("visualized_result_fg.png", vis_im);
std::cout << "Visualized result save in ./visualized_result_replaced_bg.jpg "
"and ./visualized_result_fg.jpg"
<< std::endl;
@@ -143,7 +143,7 @@ void TrtInfer(const std::string& model_dir, const std::string& image_file,
auto vis_im = fastdeploy::vision::VisMatting(im, res);
auto vis_im_with_bg = fastdeploy::vision::SwapBackground(im, bg, res);
cv::imwrite("visualized_result.jpg", vis_im_with_bg);
cv::imwrite("visualized_result_fg.jpg", vis_im);
cv::imwrite("visualized_result_fg.png", vis_im);
std::cout << "Visualized result save in ./visualized_result_replaced_bg.jpg "
"and ./visualized_result_fg.jpg"
<< std::endl;