CRNN torch转onnx的维度问题

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.83k stars 2.26k forks source link

Open ASHLEYDX opened 3 years ago

ASHLEYDX commented 3 years ago

将ocr-lstm.pth 利用torch转onnx后第二维维度是1，但发现代码中用的onnx模型第二维是3，在做框架转换时还需要做什么其他处理吗

ASHLEYDX commented 3 years ago

发现1，3分别对应灰度图和彩色图，想请问下对于彩色图该如何训练呢？

ZZHHogan commented 3 years ago

发现1，3分别对应灰度图和彩色图，想请问下对于彩色图该如何训练呢？

你好请问这个问题后面解决了吗？