WenmuZhou / PytorchOCR

基于Pytorch的OCR工具库,支持常用的文字检测和识别算法
1.39k stars 307 forks source link

onnx2trt #286

Closed txy00001 closed 3 months ago

txy00001 commented 10 months ago

你好,我用这个框架顺利转成功了onnx,并且简化了onnx-sim,但是在进行onnx2trt时,发生了错误: 1

报错信息如下: [6] Invalid Node - /head/ctc_encoder/encoder_reshape/If /head/ctc_encoder/encoder_reshape/If_OutputLayer: IIfConditionalOutputLayer inputs must have the same shape. Shapes are [-1,120,-1] and [-1,120,-1,-1]. [01/15/2024-17:21:07] [E] Failed to parse onnx file [01/15/2024-17:21:07] [I] Finished parsing network model. Parse time: 0.0395624 [01/15/2024-17:21:07] [E] Parsing model failed [01/15/2024-17:21:07] [E] Failed to create engine from model or file. [01/15/2024-17:21:07] [E] Engine set up failed 换了两个模型都报同样的错误,该如何处理呢?

WenmuZhou commented 10 months ago

不清楚哦