paddlelite在树莓派上部署百度飞桨的口罩模型

armlinuxV7 树莓派4B+ python3.7
在树莓派上部署百度飞桨口罩预测模型(python):代码如下
from paddlelite.lite import *
from PIL import Image
import numpy
config = MobileConfig()
config.set_model_from_file("./mask_detector_opt2.nb")
predictor = create_paddle_predictor(config)
image = Image.open(’./kouzhao.png’)
resized_image = image.resize((224, 224), Image.BILINEAR)
image_data = np.array(resized_image).flatten().tolist()
input_tensor = predictor.get_input(0)
input_tensor.resize([1, 3, 224, 224])
input_tensor.set_float_data(image_data)
上面的代码运行到正常,但是加了下面这个就会出错
predictor.run()
output_tensor = predictor.get_output(0)
print(output_tensor.shape())
print(output_tensor.float_data()[:10])

报错信息:
[F 8/27 10:52:57. 9 …addle-Lite/lite/kernels/arm/fc_compute.h:102 ReInitWhenNeeded] Check failed: k_ == w_dims[0]: 1568!==512
Backend terminated or disconnectedc.Fatal Python error: Aborted

还有个问题就是,百度的口罩预测模型包含俩个模型,除了mask_detector还有一个pyramidbox_lite,此处用了mask_detector 如果要用另一个pyramidbox_lite,是不是只要把上面代码改.nb文件的名称就好了

还有百度飞桨的口罩预测模型用paddlelite python版本如何设置缩放因子 shrink这些参数呢 我看飞桨的python Demo里面没有讲(https://github.com/PaddlePaddle/Paddle-Lite/blob/31572a495150521fd91bbb3ffe0fc03b7f982561/lite/demo/python/mobilenetv1_light_api.py)

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值