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

Modify infer seg demo

Browse Source
This commit is contained in:
jiangjiajun
2022-10-02 07:22:11 +00:00
parent e643743a9d 4de832bc37
commit e55cdec486
3 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
examples/vision/segmentation/paddleseg/cpp/infer.cc
View File

@@ -83,8 +83,11 @@ void TrtInfer(const std::string& model_dir, const std::string& image_file) {
auto option = fastdeploy::RuntimeOption();
option.UseGpu();
option.UseTrtBackend();
<<<<<<< HEAD
// option.SetTrtInputShape("x", {1, 3, 256, 256}, {1, 3, 1024, 1024},
// {1, 3, 2048, 2048});
=======
>>>>>>> 4de832bc37b1f9c44e4956db94c2605cd34e021b
auto model = fastdeploy::vision::segmentation::PaddleSegModel(
model_file, params_file, config_file, option);