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

[examples] Update pptinypose_infer.cc (#984)

Browse Source 
* Update pptinypose_infer.cc

* Update det_keypoint_unite_infer.cc

* Update pptinypose_infer.cc
This commit is contained in:
DefTruth
2022-12-28 10:47:26 +08:00
committed by GitHub
parent 1da821a871
commit 4f2879af81
2 changed files with 8 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
examples/vision/keypointdetection/det_keypoint_unite/cpp/det_keypoint_unite_infer.cc
View File

@@ -55,10 +55,9 @@ void CpuInfer(const std::string& det_model_dir,
} else { } else {
std::cout << "TinyPose Prediction Done!" << std::endl; std::cout << "TinyPose Prediction Done!" << std::endl;
} }
// 输出预测框结果
std::cout << res.Str() << std::endl; std::cout << res.Str() << std::endl;
// 可视化预测结果
auto vis_im = auto vis_im =
fastdeploy::vision::VisKeypointDetection(im, res, 0.2); fastdeploy::vision::VisKeypointDetection(im, res, 0.2);
cv::imwrite("vis_result.jpg", vis_im); cv::imwrite("vis_result.jpg", vis_im);
@@ -104,10 +103,9 @@ void KunlunXinInfer(const std::string& det_model_dir,
} else { } else {
std::cout << "TinyPose Prediction Done!" << std::endl; std::cout << "TinyPose Prediction Done!" << std::endl;
} }
// 输出预测框结果
std::cout << res.Str() << std::endl; std::cout << res.Str() << std::endl;
// 可视化预测结果
auto vis_im = auto vis_im =
fastdeploy::vision::VisKeypointDetection(im, res, 0.2); fastdeploy::vision::VisKeypointDetection(im, res, 0.2);
cv::imwrite("vis_result.jpg", vis_im); cv::imwrite("vis_result.jpg", vis_im);
@@ -153,10 +151,9 @@ void GpuInfer(const std::string& det_model_dir,
} else { } else {
std::cout << "TinyPose Prediction Done!" << std::endl; std::cout << "TinyPose Prediction Done!" << std::endl;
} }
// 输出预测框结果
std::cout << res.Str() << std::endl; std::cout << res.Str() << std::endl;
// 可视化预测结果
auto vis_im = auto vis_im =
fastdeploy::vision::VisKeypointDetection(im, res, 0.2); fastdeploy::vision::VisKeypointDetection(im, res, 0.2);
cv::imwrite("vis_result.jpg", vis_im); cv::imwrite("vis_result.jpg", vis_im);
@@ -210,10 +207,9 @@ void TrtInfer(const std::string& det_model_dir,
} else { } else {
std::cout << "TinyPose Prediction Done!" << std::endl; std::cout << "TinyPose Prediction Done!" << std::endl;
} }
// 输出预测关键点结果
std::cout << res.Str() << std::endl; std::cout << res.Str() << std::endl;
// 可视化预测结果
auto vis_im = auto vis_im =
fastdeploy::vision::VisKeypointDetection(im, res, 0.2); fastdeploy::vision::VisKeypointDetection(im, res, 0.2);
cv::imwrite("vis_result.jpg", vis_im); cv::imwrite("vis_result.jpg", vis_im);

12
examples/vision/keypointdetection/tiny_pose/cpp/pptinypose_infer.cc
View File

@@ -42,10 +42,9 @@ void CpuInfer(const std::string& tinypose_model_dir,
} else { } else {
std::cout << "TinyPose Prediction Done!" << std::endl; std::cout << "TinyPose Prediction Done!" << std::endl;
} }
// 输出预测框结果
std::cout << res.Str() << std::endl; std::cout << res.Str() << std::endl;
// 可视化预测结果
auto tinypose_vis_im = auto tinypose_vis_im =
fastdeploy::vision::VisKeypointDetection(im, res, 0.5); fastdeploy::vision::VisKeypointDetection(im, res, 0.5);
cv::imwrite("tinypose_vis_result.jpg", tinypose_vis_im); cv::imwrite("tinypose_vis_result.jpg", tinypose_vis_im);
@@ -75,10 +74,9 @@ void KunlunXinInfer(const std::string& tinypose_model_dir,
} else { } else {
std::cout << "TinyPose Prediction Done!" << std::endl; std::cout << "TinyPose Prediction Done!" << std::endl;
} }
// 输出预测框结果
std::cout << res.Str() << std::endl; std::cout << res.Str() << std::endl;
// 可视化预测结果
auto tinypose_vis_im = auto tinypose_vis_im =
fastdeploy::vision::VisKeypointDetection(im, res, 0.5); fastdeploy::vision::VisKeypointDetection(im, res, 0.5);
cv::imwrite("tinypose_vis_result.jpg", tinypose_vis_im); cv::imwrite("tinypose_vis_result.jpg", tinypose_vis_im);
@@ -110,10 +108,9 @@ void GpuInfer(const std::string& tinypose_model_dir,
} else { } else {
std::cout << "TinyPose Prediction Done!" << std::endl; std::cout << "TinyPose Prediction Done!" << std::endl;
} }
// 输出预测框结果
std::cout << res.Str() << std::endl; std::cout << res.Str() << std::endl;
// 可视化预测结果
auto tinypose_vis_im = auto tinypose_vis_im =
fastdeploy::vision::VisKeypointDetection(im, res, 0.5); fastdeploy::vision::VisKeypointDetection(im, res, 0.5);
cv::imwrite("tinypose_vis_result.jpg", tinypose_vis_im); cv::imwrite("tinypose_vis_result.jpg", tinypose_vis_im);
@@ -145,10 +142,9 @@ void TrtInfer(const std::string& tinypose_model_dir,
} else { } else {
std::cout << "TinyPose Prediction Done!" << std::endl; std::cout << "TinyPose Prediction Done!" << std::endl;
} }
// 输出预测框结果
std::cout << res.Str() << std::endl; std::cout << res.Str() << std::endl;
// 可视化预测结果
auto tinypose_vis_im = auto tinypose_vis_im =
fastdeploy::vision::VisKeypointDetection(im, res, 0.5); fastdeploy::vision::VisKeypointDetection(im, res, 0.5);
cv::imwrite("tinypose_vis_result.jpg", tinypose_vis_im); cv::imwrite("tinypose_vis_result.jpg", tinypose_vis_im);