From 4361d2021ed9b44b80bcd7d631bba2d7c585b565 Mon Sep 17 00:00:00 2001 From: huangjianhui <852142024@qq.com> Date: Fri, 9 Sep 2022 13:20:33 +0800 Subject: [PATCH] Update ppdet example code (#210) --- .../vision/detection/paddledetection/cpp/infer_faster_rcnn.cc | 2 ++ examples/vision/detection/paddledetection/cpp/infer_picodet.cc | 3 +++ examples/vision/detection/paddledetection/cpp/infer_ppyolo.cc | 2 ++ examples/vision/detection/paddledetection/cpp/infer_ppyoloe.cc | 3 +++ examples/vision/detection/paddledetection/cpp/infer_yolov3.cc | 2 ++ examples/vision/detection/paddledetection/cpp/infer_yolox.cc | 3 +++ 6 files changed, 15 insertions(+) diff --git a/examples/vision/detection/paddledetection/cpp/infer_faster_rcnn.cc b/examples/vision/detection/paddledetection/cpp/infer_faster_rcnn.cc index 7bd7fd91c..6471b457e 100644 --- a/examples/vision/detection/paddledetection/cpp/infer_faster_rcnn.cc +++ b/examples/vision/detection/paddledetection/cpp/infer_faster_rcnn.cc @@ -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; diff --git a/examples/vision/detection/paddledetection/cpp/infer_picodet.cc b/examples/vision/detection/paddledetection/cpp/infer_picodet.cc index 19c2a6837..e4164f9ef 100644 --- a/examples/vision/detection/paddledetection/cpp/infer_picodet.cc +++ b/examples/vision/detection/paddledetection/cpp/infer_picodet.cc @@ -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; diff --git a/examples/vision/detection/paddledetection/cpp/infer_ppyolo.cc b/examples/vision/detection/paddledetection/cpp/infer_ppyolo.cc index a111e70f5..47946c8fd 100644 --- a/examples/vision/detection/paddledetection/cpp/infer_ppyolo.cc +++ b/examples/vision/detection/paddledetection/cpp/infer_ppyolo.cc @@ -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; diff --git a/examples/vision/detection/paddledetection/cpp/infer_ppyoloe.cc b/examples/vision/detection/paddledetection/cpp/infer_ppyoloe.cc index ec01d3914..044c8eaa7 100644 --- a/examples/vision/detection/paddledetection/cpp/infer_ppyoloe.cc +++ b/examples/vision/detection/paddledetection/cpp/infer_ppyoloe.cc @@ -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; diff --git a/examples/vision/detection/paddledetection/cpp/infer_yolov3.cc b/examples/vision/detection/paddledetection/cpp/infer_yolov3.cc index 54571b8d2..5945e86f2 100644 --- a/examples/vision/detection/paddledetection/cpp/infer_yolov3.cc +++ b/examples/vision/detection/paddledetection/cpp/infer_yolov3.cc @@ -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; diff --git a/examples/vision/detection/paddledetection/cpp/infer_yolox.cc b/examples/vision/detection/paddledetection/cpp/infer_yolox.cc index 8eb9b4224..4b03f86fd 100644 --- a/examples/vision/detection/paddledetection/cpp/infer_yolox.cc +++ b/examples/vision/detection/paddledetection/cpp/infer_yolox.cc @@ -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;