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

Update docs

Browse Source
This commit is contained in:
felixhjh
2023-02-08 16:33:12 +00:00
parent bc3a7ce8fe
commit 844d80ecae
61 changed files with 1117 additions and 2519 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
examples/vision/segmentation/paddleseg/cpu-gpu/cpp/infer.cc
View File

@@ -85,7 +85,14 @@ void TrtInfer(const std::string& model_dir, const std::string& image_file) {
auto option = fastdeploy::RuntimeOption();
option.UseGpu();
option.UseTrtBackend();
auto model = fastdeploy::vision::segmentation::PaddleSegModel(
// If use original Tensorrt, not Paddle-TensorRT,
// comment the following two lines
option.EnablePaddleToTrt();
option.EnablePaddleTrtCollectShape();
option.SetTrtInputShape("x", {1, 3, 256, 256}, {1, 3, 1024, 1024},
{1, 3, 2048, 2048})
auto model = fastdeploy::vision::segmentation::PaddleSegModel(
model_file, params_file, config_file, option);
if (!model.Initialized()) {