关于mnn模型输出的数据杂乱无章问题

这俩天顺便用了阿里的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)

关于mnn模型输出的数据杂乱无章问题相关推荐

多输出模型实例的数据加载
多输出模型实例的数据加载相关的数据集放在C:/Users/Administrator/data/moc import tensorflow as tf from tensorflow import ...
14、yolov5-6中数据预处理、模型输出nms单独计算、onnxruntime的gpu版本前向推理
1.数据输入预处理.模型输出nms等 import os import torch import time import math import cv2 import numpy as np impo ...
MIKE水动力笔记7_实测数据与模型输出结果的拟合对比
本文目录前言 Step 1 拟合对比前的准备工作 Step 2 从模型输出结果dfsu文件提取出站位点处的模拟潮位dfs0文件 Step 3 将两个dfs0文件插进绘图板 Step 4 对图面进行必 ...
【板栗糖GIS】三维建模—倾斜航飞OSGB数据可以转成skp模型输出吗？
[板栗糖GIS]三维建模-倾斜航飞OSGB数据可以转成skp模型输出吗? 回答: sketchup不能直接导入航飞osgb文件,可以将osgb转换为obj或者fbx导入3dmax软件,并将文件导出为3 ...
ML之xgboost：基于xgboost(5f-CrVa)算法对HiggsBoson数据集(Kaggle竞赛)训练实现二分类预测(基于训练好的模型进行新数据预测)
ML之xgboost:基于xgboost(5f-CrVa)算法对HiggsBoson数据集(Kaggle竞赛)训练实现二分类预测(基于训练好的模型进行新数据预测) 目录输出结果设计思路核心代码 ...
Keras之DNN：利用DNN算法【Input(8)→12+8(relu)→O(sigmoid)】利用糖尿病数据集训练、评估模型(利用糖尿病数据集中的八个参数特征预测一个0或1结果)
Keras之DNN:利用DNN算法[Input(8)→12+8(relu)→O(sigmoid)]利用糖尿病数据集训练.评估模型(利用糖尿病数据集中的八个参数特征预测一个0或1结果) 目录输出结果 ...
机器学习数据模型_使用PyCaret将机器学习模型运送到数据—第二部分
机器学习数据模型 My previous post Machine Learning in SQL using PyCaret 1.0 provided details about integrat ...
模型算法_数据分析师，要懂多少算法模型？
随着数据分析岗位招聘越来越内卷,问"你用过/建过什么模型"的也越来越多.这个问题很容易给人"面试造航母,工作拧螺丝"的感觉.实际工作中,真的要搞那么多模型??? ...
水环境模型与大数据技术融合研究
点击上方蓝字关注我们水环境模型与大数据技术融合研究马金锋1, 饶凯锋1, 李若男1,2, 张京1, 郑华1,2 1 中国科学院生态环境研究中心城市与区域生态国家重点实验室,北京 100085 2 ...

关于mnn模型输出的数据杂乱无章问题

关于mnn模型输出的数据杂乱无章问题相关推荐

最新文章

热门文章