当前位置:网站首页>关于mnn模型输出的数据杂乱无章问题
关于mnn模型输出的数据杂乱无章问题
2022-08-03 23:45:00 【氵文大师】
这俩天顺便用了阿里的MNN:
https://github.com/alibaba/MNN
还挺好用的,就是不知道带有opencl后端的引擎怎么用Python API调用
我遇到一个小坑,输出的数据是这样杂乱无章的:
而不是这样的:
反复debug,发现:
原来是模型的输出不能直接 getData()
output_tensor = interpreter.getSessionOutput(session) # 获得模型的输出
tmp_output = MNN.Tensor((1, 2, 224, 224), # 用来做输出的临时变量
MNN.Halide_Type_Float,
np.ones([1, 2, 224, 224]).astype(np.float32),
MNN.Tensor_DimensionType_Caffe)
output_tensor.copyToHostTensor(tmp_output) # 将模型的输出给 tmp_output 变量
x = tmp_output.getNumpyData()[0] # 获取 numpy 格式的数据
这段代码没啥问题,但要把最后一行:
x = tmp_output.getNumpyData()[0] # 获取 numpy 格式的数据
替换为:
x = output_tensor.getNumpyData()[0] # 获取 numpy 格式的数据
就会变成杂乱无章的数据,可能是数据先行后列之类的出问题了??
他俩都是 MNN.Tensor
的数据类型
所以我感觉,MNN模型输出的Tensor,要先转化为对应的格式MNN.Tensor_DimensionType_Caffe
,才能打印出来
或者说这一步就是转化数据格式的:
tmp_output = MNN.Tensor((1, 2, 224, 224), # 用来做输出的临时变量
MNN.Halide_Type_Float,
np.ones([1, 2, 224, 224]).astype(np.float32),
MNN.Tensor_DimensionType_Caffe)
边栏推荐
- 二叉搜索树解决落叶问题
- (PC+WAP)织梦模板螺钉手柄类网站
- ros mavros stereo读取rosbag并记录IMU和图片到文件夹
- Scala基础【正则表达式、框架式开发原则】
- V8中的快慢数组(附源码、图文更易理解)
- Go编译原理系列7(Go源码调试)
- leetcode/子串中不能有重复字符的最长子串
- 全球首款量产,获定点最多!这家AVP Tier1如何实现领跑?
- The longest substring that cannot have repeating characters in a leetcode/substring
- FinClip, help smart TV more imagination
猜你喜欢
随机推荐
双目IMU标定kalibr
leetcode/子串中不能有重复字符的最长子串
redis持久化方式
CAS:178744-28-0,mPEG-DSPE,DSPE-mPEG,甲氧基-聚乙二醇-磷脂酰乙醇胺供应
Creo 9.0二维草图的诊断:加亮开放端点
Graph-node:创建一个新的subgraph
电子邮件安全或面临新威胁!
小身材有大作用——光模块基础知识(一)
软测人每个阶段的薪资待遇,快来康康你能拿多少?
The super perfect layout has shortcut keys and background replacement
Internship: Upload method for writing excel sheet (import)
响应式织梦模板除尘器类网站
V8中的快慢数组(附源码、图文更易理解)
冰河又一MySQL力作出版(文末送书)!!
Free自由协议系统开发
ML之interpret:基于titanic泰坦尼克是否获救二分类预测数据集利用interpret实现EBC模型可解释性之全局解释/局部解释案例
逆波兰表达式求值
ML's yellowbrick: A case of interpretability (threshold map) for LoR logistic regression model using yellowbrick based on whether Titanic was rescued or not based on the two-class prediction dataset
689. 三个无重叠子数组的最大和
Creo9.0 绘制中心线