Toybrick

人工智能开发系列(7) OPENPOSE开发与实现

bill

新手上路

积分
35
楼主
发表于 2019-6-11 06:47:45 | 显示全部楼层
您好,我转换模型的时候出现如下错误,请问如何解决,docker 环境下执行的转换
root@89ee00158270:/home/OpenPose-Multi-Person# python3 rknn_transfer.py
--> config model
done
--> Loading model
E Deprecated caffe input usage, please change it to input layer.
E Catch exception when loading caffe model: ./pose/coco/pose_deploy_linevec.prototxt!
T Traceback (most recent call last):
T   File "rknn/api/rknn_base.py", line 281, in rknn.api.rknn_base.RKNNBase.load_caffe
T   File "rknn/base/RKNNlib/converter/caffeloader.py", line 977, in rknn.base.RKNNlib.converter.caffeloader.CaffeLoader.load
T   File "rknn/base/RKNNlib/converter/caffeloader.py", line 746, in rknn.base.RKNNlib.converter.caffeloader.CaffeLoader.parse_net_param
T   File "rknn/base/RKNNlib/RKNNlog.py", line 105, in rknn.base.RKNNlib.RKNNlog.RKNNLog.e
T ValueError: Deprecated caffe input usage, please change it to input layer.
Load model failed!


回复

使用道具 举报

bill

新手上路

积分
35
沙发
发表于 2019-6-11 08:23:28 | 显示全部楼层
bill 发表于 2019-6-11 06:47
您好,我转换模型的时候出现如下错误,请问如何解决,docker 环境下执行的转换
root@89ee00158270:/home/Op ...

我的已解决了,pose_deploy_linevec.prototxt 模型文件版本低的问题
回复

使用道具 举报

bill

新手上路

积分
35
板凳
发表于 2019-6-12 07:26:36 | 显示全部楼层
在docker 环境执行了这例子。但一个姿态都没有识别出来,output 打印基本全是【0,。。。。0】。
docker 环境为跑这个例子,源码主要改动点:
a.模型转换中将 pre_compile=False ,原来True 转换报错说,不支持pre_compile
b. 将从摄像图输入,改为输入图片或视屏
c. output = output.reshape(1, 57, 46, 46) 修改为 output = output.reshape(1, 57, 28, 28) 因为output size 是44688 不能reshape 为 (1, 57, 46, 46)
回复

使用道具 举报

bill

新手上路

积分
35
地板
发表于 2019-6-12 23:00:26 | 显示全部楼层
早上这个问题也解决了,是由于修改prototxt文件参数,dim 228 要修改368
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表