apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-12-24 13:28:13 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Update infer_ppyoloe.cc

Browse Source
This commit is contained in:
Jason
2022-09-15 14:03:40 +08:00
committed by GitHub
parent 09788d265b
commit f77f558d81
1 changed files with 0 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/vision/detection/paddledetection/cpp/infer_ppyoloe.cc
View File

@@ -83,8 +83,6 @@ void TrtInfer(const std::string& model_dir, const std::string& image_file) {
auto option = fastdeploy::RuntimeOption();
option.UseGpu();
option.UseTrtBackend();
option.SetTrtInputShape("image", {1, 3, 640, 640});
option.SetTrtInputShape("scale_factor", {1, 2});
auto model = fastdeploy::vision::detection::PPYOLOE(model_file, params_file,
config_file, option);
if (!model.Initialized()) {