当前位置:网站首页>关于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)
边栏推荐
- Software testing is seriously involution, how to improve your competitiveness?
- Kotlin - 扩展函数和运算符重载
- P1449 后缀表达式
- Interpretation of ML: A case of global interpretation/local interpretation of EBC model interpretability based on titanic titanic rescued binary prediction data set using interpret
- rsync basic usage
- 射频芯片(RFIC)的协议之5G及其调制
- Prometheus监控Harbor(二进制版)
- 小身材有大作用——光模块基础知识(一)
- Creo 9.0二维草图的诊断:重叠几何
- FPGA按键消抖+蜂鸣器
猜你喜欢

一文搞定 SQL Server 执行计划

Jmeter-断言

The world's first mass production, with the most fixed points!How does this AVP Tier1 lead?

FinClip, help smart TV more imagination

Redis persistence method

全球首款量产,获定点最多!这家AVP Tier1如何实现领跑?

射频芯片(RFIC)的协议之5G及其调制

重新认识浏览器的渲染过程

BMN: Boundary-Matching Network for Temporal Action Proposal Generation Reading Notes

3D Semantic Segmentation - 2DPASS
随机推荐
P1449 后缀表达式
牛客2022 暑期多校3 H Hacker(SAM + 线段树查询区间内部最大子段和)
End-to-End Lane Marker Detection via Row-wise Classification
(PC+WAP)织梦模板不锈钢类网站
代码随想录笔记_动态规划_416分割等和子集
AOSP CameraLatencyHistogram的原理与使用
Why Flutter Flutter of tutorials is the best choice for business?
射频芯片(RFIC)的协议之5G及其调制
Jmeter-参数化
超级完美版布局有快捷键,有背景置换
ts用法大全
用两个栈模拟队列
数据分析知识点搜集(纯粹的搜集)
【LeetCode】最长公共子序列(动态规划)
Creo 9.0二维草图的诊断:加亮开放端点
OPC UA 与IEC61499 深度融合(1)
Deep integration of OPC UA and IEC61499 (1)
The longest substring that cannot have repeating characters in a leetcode/substring
1067 Sort with Swap(0, i)
HNUCM 您好中国