Update ppdet example code (#210)

This commit is contained in:
huangjianhui
2022-09-09 13:20:33 +08:00
committed by GitHub
parent ea27f089a6
commit 4361d2021e
6 changed files with 15 additions and 0 deletions

View File

@@ -40,6 +40,7 @@ void CpuInfer(const std::string& model_dir, const std::string& image_file) {
return;
}
std::cout << res.Str() << std::endl;
auto vis_im = fastdeploy::vision::Visualize::VisDetection(im_bak, res, 0.5);
cv::imwrite("vis_result.jpg", vis_im);
std::cout << "Visualized result saved in ./vis_result.jpg" << std::endl;
@@ -68,6 +69,7 @@ void GpuInfer(const std::string& model_dir, const std::string& image_file) {
return;
}
std::cout << res.Str() << std::endl;
auto vis_im = fastdeploy::vision::Visualize::VisDetection(im_bak, res, 0.5);
cv::imwrite("vis_result.jpg", vis_im);
std::cout << "Visualized result saved in ./vis_result.jpg" << std::endl;
@@ -99,6 +101,7 @@ void TrtInfer(const std::string& model_dir, const std::string& image_file) {
return;
}
std::cout << res.Str() << std::endl;
auto vis_im = fastdeploy::vision::Visualize::VisDetection(im_bak, res, 0.5);
cv::imwrite("vis_result.jpg", vis_im);
std::cout << "Visualized result saved in ./vis_result.jpg" << std::endl;