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

Add detection and segmentation examples for Ascend deployment

Browse Source
This commit is contained in:
yunyaoXYY
2022-12-27 07:40:46 +00:00
parent 76a876406e
commit 090c3a68b4
27 changed files with 280 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
examples/vision/detection/yolov7/cpp/infer_paddle_model.cc
View File

@@ -31,7 +31,7 @@ void InitAndInfer(const std::string& model_dir, const std::string& image_file,
auto im = cv::imread(image_file);
fastdeploy::vision::DetectionResult res;
if (!model.Predict(&im, &res)) {
if (!model.Predict(im, &res)) {
std::cerr << "Failed to predict." << std::endl;
return;
}
@@ -68,7 +68,9 @@ int main(int argc, char* argv[]) {
option.UseTrtBackend();
} else if (flag == 2) {
option.UseKunlunXin();
}
} else if (flag == 3) {
option.UseAscend();
}
std::string model_dir = argv[1];
std::string test_image = argv[2];