apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-05 00:33:03 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Update infer_picodet.cc

Browse Source
This commit is contained in:
Jason
2022-09-15 14:03:23 +08:00
committed by GitHub
parent ec835a7d85
commit 09788d265b
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_picodet.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, 320, 320});
option.SetTrtInputShape("scale_Factor", {1, 2});
auto model = fastdeploy::vision::detection::PicoDet(model_file, params_file,
config_file, option);
if (!model.Initialized()) {