无法将 onnx 模型转换为 TRT 模型
作者: k-khosravi创建于 2023年12月16日更新于 2024年8月7日
我已经将此模型"20180402-114759"转换为用于人脸识别的 ONNX 格式,使用了 onnx==1.14.1。当我想将 ONNX 模型转换为 TRT 时,会出现以下错误: " [E] Error[4]: [graphShapeAnalyzer.cpp::processCheck::862] 错误代码 4: 内部错误 (StatefulPartitionedCall/inception_resnet_v1/Conv2d_2a_3x3/Conv2D: 卷积/反卷积输出的空间维度不能为负 (轴 2 的构建时输出维度为 -2)) [12/16/2023-15:52:45] [E] 引擎无法从网络中创建 [12/16/2023-15:52:45] [E] 创建引擎失败 [12/16/2023-15:52:45] [E] 无法从模型或文件中创建引擎。 [12/16/2023-15:52:45] [E] 引擎设置失败 &&&& FAILED TensorRT.trtexec [TensorRT v8601] # ...." 我使用 CUDA 12.1 的 TensorRT 8.6。文档中提到,对于 TensorRT 8.6,onnxruntime 应为 1.15 或更高版本。但是,我无法使用 onnx==1.15.0 将 tensorflow 模型转换为 ONNX 模型,并且遇到了一些错误。如何解决此问题,以成功将 tensorflow 模型转换为 ONNX 模型,然后转换为 TRT 模型?
内容来源: davidsandberg/facenet