当前位置:网站首页>关于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)
边栏推荐
- Free自由协议系统开发
- 七夕活动浪漫上线,别让网络拖慢和小姐姐的开黑时间
- [2022安恒夏令营] 5个小题
- [Paper Reading] TRO 2021: Fail-Safe Motion Planning for Online Verification of Autonomous Vehicles Using Conve
- End-to-End Lane Marker Detection via Row-wise Classification
- FinClip, help smart TV more imagination
- OpenCV 图像拼接
- 栈的压入、弹出序列
- Unity 截取3D图像 与 画中画PIP的实现
- 逆波兰表达式求值
猜你喜欢

Pytest learn-setup/teardown

Jmeter-断言

【OpenCV图像处理】 图像拼接技术

Minimized installation of debian11

Another MySQL masterpiece published by Glacier (send the book at the end of the article)!!

SPOJ 2774 Longest Common Substring(两串求公共子串 SAM)

rosbridge-WSL2 && carla-win11

V8中的快慢数组(附源码、图文更易理解)

Pytest学习-setup/teardown

单例模式使用饿汉式和懒汉式创建一定安全?很多人不知
随机推荐
Minimized installation of debian11
Unity 截取3D图像 与 画中画PIP的实现
Pytest learn-setup/teardown
简单了解下 TCP,学习握手和挥手以及各种状态到底是怎么样的
rosbridge-WSL2 && carla-win11
Internship: Upload method for writing excel sheet (import)
【LeetCode】最长公共子序列(动态规划)
响应式织梦模板餐饮酒店类网站
AOSP CameraLatencyHistogram的原理与使用
禾匠编译错误记录
HNUCM 您好中国
Pytest学习-setup/teardown
ros mavros stereo读取rosbag并记录IMU和图片到文件夹
Prometheus监控Harbor(二进制版)
Why Flutter Flutter of tutorials is the best choice for business?
走迷宫 BFS
二叉搜索树解决落叶问题
ML之yellowbrick:基于titanic泰坦尼克是否获救二分类预测数据集利用yellowbrick对LoR逻辑回归模型实现可解释性(阈值图)案例
libnet
七夕活动浪漫上线,别让网络拖慢和小姐姐的开黑时间