search

DayBreak-u / chineseocr_lite

超轻量级中文ocr，支持竖排文字识别, 支持ncnn、mnn、tnn推理 ( dbnet(1.8M) + crnn(2.5M) + anglenet(378KB)) 总模型仅4.7M
GNU General Public License v2.0
11.84k stars 2.26k forks source link

onnxruntime Error #376

Open ekils opened 3 years ago

ekils commented 3 years ago

你好想請問, 我將訓練好的crnn模型轉onnx後, 模型結構如下：

截圖 2021-09-11 下午4 30 10

可是運行predict後報報如下圖：

截圖 2021-09-11 下午4 32 00

說input的尺寸不符, channel ＝3 那個還能理解, 頂多我丟進去的iamge改成grey, 可是的277的寬度要怎辦？ 我依照版主提供的轉onnx 文章, 裡面提到 ： inputs = torch.randn(1, 1, 32, 277).to(self.device) 看來onnx 是吃死的寬度？ 還是哪裡理解有誤, 請幫忙解惑。謝謝

ekils commented 3 years ago

我在產生onnx時增加了 dynamic_axes 的參數, 變成動態的input, 不過連simplier 都會報錯, 直接跑推論也會錯誤