PaddlePaddle / FastDeploy

⚡️An Easy-to-use and Fast Deep Learning Model Deployment Toolkit for ☁️Cloud 📱Mobile and 📹Edge. Including Image, Video, Text and Audio 20+ main stream scenarios and 150+ SOTA models with end-to-end optimization, multi-platform and multi-framework support.
https://www.paddlepaddle.org.cn/fastdeploy
Apache License 2.0
2.83k stars 444 forks source link

如何给tensorrt添加shape #2144

Open guoyunqingyue opened 11 months ago

guoyunqingyue commented 11 months ago

使用c#api的fastdeploy,如何使用TensorRT? 目前使用的代码: RuntimeOption runtimeoption = new RuntimeOption(); runtimeoption.UseGpu(); runtimeoption.UseTrtBackend(); runtimeoption.EnablePaddleToTrt(); runtimeoption.EnablePaddleTrtCollectShape(); fastdeploy.vision.classification.PaddleClasModel model = new fastdeploy.vision.classification.PaddleClasModel(model_file, params_file, config_file, runtimeoption, ModelFormat.PADDLE); 运行会报错:

捕获 显然是需要给tensorrt添加shape,但c#中没有如同c++中的SetTrtshape接口

rainyfly commented 5 months ago

已收到反馈,后续可能需要补齐接口