- 微信
- 微博
  
  分享文章到微博
- 复制链接
  
  复制链接到剪贴板

使用小藤实现手写汉字识别

张辉发表于 2023/09/01 07:35:11 2023/09/01

【摘要】使用小藤实现手写汉字识别

代码仓地址：https://gitee.com/ascend/ascend_community_projects/tree/310B/ChineseOCR

由于这里使用的是paddlepaddle，为了不干扰其他编译环境，笔者决定创建一个新的conda环境：

cd ~/ascend_community_projects/ChineseOCR

conda create -n chineseocr python=3.9

conda activate chineseocr

pip install paddle2onnx==0.9.5 -i https://pypi.tuna.tsinghua.edu.cn/simple

没有0.9.5版本。换成0.9.2试试：

pip install paddle2onnx==0.9.2 -i https://pypi.tuna.tsinghua.edu.cn/simple

应该是新的pip环境少了什么包。

换成1.0.5试试：

pip install paddle2onnx==1.0.5 -i https://pypi.tuna.tsinghua.edu.cn/simple

貌似还是版本高的好一点。

pip install paddlepaddle==2.3.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

咦，飞桨你下线了吗？

去飞桨官网看看是怎么回事：

打开

要装2.5.1版本：

python -m pip install paddlepaddle==2.5.1 -i https://pypi.tuna.tsinghua.edu.cn/simple

或者换成老版本：

python -m pip install paddlepaddle==2.3.2 -i https://pypi.tuna.tsinghua.edu.cn/simple

只能说，飞桨貌似没有linux arm的安装包。

下载模型文件：

cd model

wget https://paddleocr.bj.bcebos.com/dygraph_v2.0/ch/ch_ppocr_server_v2.0_rec_infer.tar

解压：

tar xvf ch_ppocr_server_v2.0_rec_infer.tar

将paddlepaddle模型转为onnx模型：

paddle2onnx --model_dir ./ch_ppocr_server_v2.0_rec_infer/ --model_filename inference.pdmodel --params_filename inference.pdiparams --save_file ./ch_ppocr_server_v2.0_rec_infer.onnx --opset_version 11 --enable_onnx_checker True

可以看到 ch_ppocr_server_v2.0_rec_infer.onnx 文件已生成。（貌似可以不用装paddlepaddle)

将onnx模型转为om离线模型

创建一个aipp.cfg文件，内容如下：

aipp_op{
    related_input_rank: 0
    crop: false
    input_format: 3
    aipp_mode: static
    csc_switch: false
    mean_chn_0: 0
    mean_chn_1: 0
    mean_chn_2: 0
    min_chn_0: 127.5
    min_chn_1: 127.5
    min_chn_2: 127.5
    var_reci_chn_0: 0.007843137718737125
    var_reci_chn_1: 0.007843137718737125
    var_reci_chn_2: 0.007843137718737125
}

缺numpy:

pip install numpy==1.19.5 -i https://pypi.tuna.tsinghua.edu.cn/simple

再来：

atc --model=ch_ppocr_server_v2.0_rec_infer.onnx --framework=5 --input_format=NCHW --input_shape="x:1,3,32,320" --output=ch_ppocr_server_v2.0_rec_infer_bs1 --soc_version=Ascend310B1 --log=error --insert_op_conf=./aipp.cfg

进入缺什么补什么的模式。。。

或者，conda环境切换回base环境，因为atc应该不需要paddlepaddle环境。（更何况paddlepaddle环境还没装起来）

conda deactivate