diff --git a/examples/vision/segmentation/paddleseg/cpu-gpu/cpp/infer.cc b/examples/vision/segmentation/paddleseg/cpu-gpu/cpp/infer.cc index af79d3204..eefeebeba 100755 --- a/examples/vision/segmentation/paddleseg/cpu-gpu/cpp/infer.cc +++ b/examples/vision/segmentation/paddleseg/cpu-gpu/cpp/infer.cc @@ -90,7 +90,7 @@ void TrtInfer(const std::string& model_dir, const std::string& image_file) { option.EnablePaddleToTrt(); option.EnablePaddleTrtCollectShape(); option.SetTrtInputShape("x", {1, 3, 256, 256}, {1, 3, 1024, 1024}, - {1, 3, 2048, 2048}) + {1, 3, 2048, 2048}); auto model = fastdeploy::vision::segmentation::PaddleSegModel( model_file, params_file, config_file, option);