当前位置:网站首页>关于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)
边栏推荐
- 图论-虚拟节点分层建图
- Deep integration of OPC UA and IEC61499 (1)
- Fluorescein-PEG-CLS, cholesterol-polyethylene glycol-fluorescein scientific research reagent
- XSLT – 服务器端概述
- 直播系统聊天技术(八):vivo直播系统中IM消息模块的架构实践
- IELTS essay writing template
- 状态机实验
- libnet
- The salary of soft testers at each stage, come to Kangkang, how much can you get?
- 免费的公共WiFi不要乱连,遭中间人攻击了吧?
猜你喜欢
随机推荐
射频芯片(RFIC)的协议之5G及其调制
Creo 9.0二维草图的诊断:重叠几何
The principle and use of AOSP CameraLatencyHistogram
YOLOv7改进之二十二:涨点神器——引入递归门控卷积(gnConv)
FinClip,助长智能电视更多想象空间
C语言实验十四 结构体
Internship: Upload method for writing excel sheet (import)
Pytest learn-setup/teardown
[2022强网杯] polydiv和gamemaster
电子邮件安全或面临新威胁!
冰河又一MySQL力作出版(文末送书)!!
HNUCM 2022年暑假ACM搜索专项练习
Go编译原理系列7(Go源码调试)
Scala basics [regular expressions, framework development principles]
rosbridge-WSL2 && carla-win11
vscode插件设置——Golang开发环境配置
代码随想录笔记_动态规划_416分割等和子集
Unity 截取3D图像 与 画中画PIP的实现
ML之yellowbrick:基于titanic泰坦尼克是否获救二分类预测数据集利用yellowbrick对LoR逻辑回归模型实现可解释性(阈值图)案例
用队列模拟实现栈



![Scala basics [regular expressions, framework development principles]](/img/be/6531f32494b8516cd7031223d14bb2.png)





