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

Update infer.cc

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

2
examples/vision/classification/paddleclas/cpp/infer.cc
View File

@@ -77,8 +77,6 @@ void TrtInfer(const std::string& model_dir, const std::string& image_file) {
auto option = fastdeploy::RuntimeOption(); auto option = fastdeploy::RuntimeOption();
option.UseGpu(); option.UseGpu();
option.UseTrtBackend(); option.UseTrtBackend();
option.SetTrtInputShape("inputs", {1, 3, 224, 224}, {1, 3, 224, 224},
{1, 3, 224, 224});
auto model = fastdeploy::vision::classification::PaddleClasModel( auto model = fastdeploy::vision::classification::PaddleClasModel(
model_file, params_file, config_file, option); model_file, params_file, config_file, option);
if (!model.Initialized()) { if (!model.Initialized()) {