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

Update SwapBackgroundSegmentation && SwapBackgroundMatting to SwapBackground

Browse Source
This commit is contained in:
felixhjh
2022-12-12 10:59:15 +00:00
parent 631593b23e
commit 37359c65f3
22 changed files with 62 additions and 113 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
examples/vision/detection/yolov5/quantize/cpp/infer.cc
View File

@@ -29,7 +29,6 @@ void InitAndInfer(const std::string& model_dir, const std::string& image_file,
assert(model.Initialized());
auto im = cv::imread(image_file);
auto im_bak = im.clone();
fastdeploy::vision::DetectionResult res;
if (!model.Predict(&im, &res)) {
@@ -39,7 +38,7 @@ void InitAndInfer(const std::string& model_dir, const std::string& image_file,
std::cout << res.Str() << std::endl;
auto vis_im = fastdeploy::vision::VisDetection(im_bak, res);
auto vis_im = fastdeploy::vision::VisDetection(im, res);
cv::imwrite("vis_result.jpg", vis_im);
std::cout << "Visualized result saved in ./vis_result.jpg" << std::endl;
}