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)