search

zjchuyp / ocr.pytorch

crnn for chinese ocr
2 stars 1 forks source link

大兄弟， 你关于opencv读取pytorch的crnn模型问题解决了吗 #1

Open dgl547437235 opened 4 years ago

dgl547437235 commented 4 years ago

我也刚好碰到这个问题，pytorch的crnn网络输出为onnx模型，然后opencv去调，opencv读取不了

amandazw commented 4 years ago

大兄弟 你opencv读成功了吗

dgl547437235 commented 4 years ago

opencv4.3版本之后就可以读取了

amandazw commented 4 years ago

opencv4.3版本之后就可以读取了

用了4.3还是报错， 报错内容： cv2.error: OpenCV(4.3.0) C:\projects\opencv-python\opencv\modules\dnn\src\layers\reshape_layer.cpp:149: error: (-215:Assertion failed) dstTotal != 0 in function 'cv::dnn::computeShapeByReshapeMask 能指点一下吗

dgl547437235 commented 4 years ago

我是自己写的crnn，改了下基础架构，导出的onnx是可以用的，链接在这里https://github.com/dgl547437235/Easy_CRNN

amandazw commented 4 years ago

我是自己写的crnn，改了下基础架构，导出的onnx是可以用的，链接在这里https://github.com/dgl547437235/Easy_CRNN

ok 感谢~我参考下