树莓派上使用paddle预训练模型

树莓派安装paddlelite; x86电脑上安装paddlehub,并将paddlehub中的预训练模型转换为paddlelite格式,使之能在树莓派上运行。

树莓派编译安装paddlelite

sudo apt install patchelf cmake
sudo update-alternatives --install /usr/bin/python python /usr/bin/python3 150

git clone https://github.com/PaddlePaddle/Paddle-Lite.git
cd Paddle-Lite
sudo ./lite/tools/build.sh \
  --build_extra=ON \
  --arm_os=armlinux \
  --arm_abi=armv7hf \
  --arm_lang=gcc \
  --build_python=ON \
  full_publish

cd build.lite.armlinux.armv7hf.gcc/inference_lite_lib.armlinux.armv7hf/python/install
sudo python3 setup.py install

具体源码编译时的参数说明,大家可参考:https://paddlepaddle.github.io/Paddle-Lite/v2.2.0/source_compile/#本地编译直接在rk3399或树莓派上编译

x86电脑上将paddlehub的预训练模型转为paddlelite格式

安装paddlehub、paddlepaddle

python -m pip install paddlehub
python -m pip install paddlepaddle

测试paddlehub预训练模型

import paddlehub as hub

senta = hub.Module(name="senta_gru")
test_text = ["这家餐厅很好吃", "这部电影真的很差劲"]

results = senta.sentiment_classify(texts=test_text, use_gpu=False, batch_size=1)

for result in results:
    print(result['text'])
    print(result['sentiment_label'])
    print(result['sentiment_key'])
    print(result['positive_probs'])
    print(result['negative_probs'])

下载opt转换工具

https://paddlepaddle.github.io/Paddle-Lite/v2.2.0/model_optimize_tool/

下载paddlehub预训练模型并转换

下载预训练模型压缩包:

hub download senta_gru

解压,放置在 saved_models/senta_gru 下,

转换模型:

./opt --model_dir=senta_gru/infer_model --valid_targets=arm --optimize_out_type=naive_buffer --optimize_out=saved_models/senta_gru

PS: 查看opt转换工具所支持的模型算子:

./opt --print_model_ops=true --valid_targets=arm --model_dir=senta_gru/infer_model

在树莓派上测试转换后的预训练模型

from paddlelite.lite import *

config = MobileConfig()
config.set_model_dir("model")

# (2) 创建predictor
predictor = create_paddle_predictor(config)
'''
......
'''

然后就出现段错误了。。。


后记

之后开始各种找原因,也想到可能是由于漏转换了什么而导致的。。。

也反复找了资料,但由于paddle还尚未稳定,甚至连官方网站里的demo都已过时,接口都已变动得消失不见了。。。

所以最终只好暂时放弃。

可能 paddlepaddle 和 paddlehub ,作为百度paddle框架的主力,迭代速度快、完成度也还行,属于比较可用。但其他的组件可能优先级就比较低了,文档也不同步,实在是造成了很大的困扰。

毕竟已经占用了一整天的时间了,还要抽空学习推荐系统,和完成我的與情预测股票项目,所以paddlelite项目就只能等百度paddle稳定了再启动吧。

file

  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 5
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

zhangsheng377

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值