树莓派4B+部署paddlelite口罩预测模型 预测程序出错
Created by: 13433583369
armlinuxV7 树莓派4B+ python3.7 在树莓派上部署口罩预测模型:代码如下 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文件的名称就好了,模型的参数shrink这些又该如何改呢。
实在没找到python实现的demo,搜了许多资料也没解决,您可以指点一下吗谢谢了