diff --git a/README_CN.md b/README_CN.md index 662e63cb0..3b4ea0047 100644 --- a/README_CN.md +++ b/README_CN.md @@ -131,18 +131,20 @@ wget https://gitee.com/paddlepaddle/PaddleDetection/raw/release/2.4/demo/0000000 // GPU/TensorRT部署参考 examples/vision/detection/paddledetection/cpp #include "fastdeploy/vision.h" -int main(int argc, char* argv[]) { - namespace vision = fastdeploy::vision; - auto model = vision::detection::PPYOLOE("ppyoloe_crn_l_300e_coco/model.pdmodel", - "ppyoloe_crn_l_300e_coco/model.pdiparams", - "ppyoloe_crn_l_300e_coco/infer_cfg.yml"); - auto im = cv::imread("000000014439.jpg"); +int main(int argc, char *argv[]) { - vision::DetectionResult res; - model.Predict(&im, &res) + namespace vision = fastdeploy::vision; + auto model = vision::detection::PPYOLOE("ppyoloe_crn_l_300e_coco/model.pdmodel", + "ppyoloe_crn_l_300e_coco/model.pdiparams", + "ppyoloe_crn_l_300e_coco/infer_cfg.yml"); + auto im = cv::imread("000000014439.jpg"); + cv::cvtColor(im,im,cv::COLOR_BGR2RGB); + vision::DetectionResult res; + model.Predict(&im, &res); - auto vis_im = vision::Visualize::VisDetection(im, res, 0.5); - cv::imwrite("vis_image.jpg", vis_im); + auto vis_im = vision::Visualize::VisDetection(im, res, 0.5); + cv::imwrite("vis_image.jpg", vis_im); + return 0; } ``` diff --git a/README_EN.md b/README_EN.md index 4b6de076a..caf5e7bf9 100644 --- a/README_EN.md +++ b/README_EN.md @@ -140,19 +140,21 @@ wget https://gitee.com/paddlepaddle/PaddleDetection/raw/release/2.4/demo/0000000 // For GPU/TensorRT deployment, please refer to examples/vision/detection/paddledetection/cpp #include "fastdeploy/vision.h" -int main(int argc, char* argv[]) { - namespace vision = fastdeploy::vision; - auto model = vision::detection::PPYOLOE("ppyoloe_crn_l_300e_coco/model.pdmodel", - "ppyoloe_crn_l_300e_coco/model.pdiparams", - "ppyoloe_crn_l_300e_coco/infer_cfg.yml"); - auto im = cv::imread("000000014439.jpg"); +int main(int argc, char *argv[]) { - vision::DetectionResult res; - model.Predict(&im, &res) + namespace vision = fastdeploy::vision; + auto model = vision::detection::PPYOLOE("ppyoloe_crn_l_300e_coco/model.pdmodel", + "ppyoloe_crn_l_300e_coco/model.pdiparams", + "ppyoloe_crn_l_300e_coco/infer_cfg.yml"); + auto im = cv::imread("000000014439.jpg"); + cv::cvtColor(im,im,cv::COLOR_BGR2RGB); + vision::DetectionResult res; + model.Predict(&im, &res); - auto vis_im = vision::Visualize::VisDetection(im, res, 0.5); - cv::imwrite("vis_image.jpg", vis_im); - } + auto vis_im = vision::Visualize::VisDetection(im, res, 0.5); + cv::imwrite("vis_image.jpg", vis_im); + return 0; +} ``` ### For more deployment models, please refer to [Vision Model Deployment Examples](examples/vision) .