PaddlePaddle / Paddle2ONNX

ONNX Model Exporter for PaddlePaddle
Apache License 2.0
736 stars 172 forks source link

PaddleOCR的模型能通过Paddle2ONNX转为ONNX吗? #232

Open YJLCV opened 3 years ago

YJLCV commented 3 years ago

我想把PaddleOCR的模型,例如ch_ppocr_mobile_v2.0_det_infer,转化为ONNX,然后再转为NCNN。所以能通过Paddle2ONNX能将其转化为ONNX,若可以,请问要怎么操作?

jiangjiajun commented 3 years ago

参考这个教程 https://aistudio.baidu.com/aistudio/projectdetail/1479970?channelType=0&channel=0

YJLCV commented 3 years ago

参考这个教程 https://aistudio.baidu.com/aistudio/projectdetail/1479970?channelType=0&channel=0

好的感谢 我先研究下

YJLCV commented 3 years ago

@jiangjiajun 您好,感谢你的链接,现在我已成功将PaddleOCR的模型,ch_ppocr_mobile_v2.0_det_infer,转化为ONNX,然后再转为NCNN了。再想请教下,如何测试自己转的NCNN模型是否正确?例如能跑个demo啥的吗,如果用ncnn的模型也能做OCR,那么应该就是转换成功了?感谢您的回复!

znsoftm commented 3 years ago

目前转成ncnn有问题,只有部分模型工作, 参见 : https://github.com/RapidOCR/RapidOCRNcnnCpp/tree/main/models

jiangjiajun commented 3 years ago

目前转成ncnn有问题,只有部分模型工作, 参见 : https://github.com/RapidOCR/RapidOCRNcnnCpp/tree/main/models

根据你这里的反馈,是目前ch_ppocr_mobile_v2.0_rec, ch_ppocr_mobile_v2.0_cls这两个不行,其余都是可以的是么

YJLCV commented 3 years ago

PaddleOCR2ONNX,用https://aistudio.baidu.com/aistudio/projectdetail/1479970?channelType=0&channel=0就行了, 然后ONNX转NCNN用这个不就行了https://convertmodel.com/