MNN icon indicating copy to clipboard operation
MNN copied to clipboard

pymnn和C++推理结果不一致

Open mutouoa opened this issue 2 years ago • 1 comments

pymnn和C++推理结果不一致

mutouoa avatar Sep 08 '22 09:09 mutouoa

@jxt1234

mutouoa avatar Sep 08 '22 09:09 mutouoa

macos平台:使用下面这两行代码,得到的类型是MNN.tensor类型,这种类型能转化成torch.tensor然后接着执行模型下面的代码吗?

    tmp_output = MNN.Tensor((1, 240, 384), MNN.Halide_Type_Float, np.ones([1, 240, 384]).astype(np.float32),
                            MNN.Tensor_DimensionType_Caffe)
    output_tensor.copyToHostTensor(tmp_output)

如果可以的话,怎么转换啊,求助!!!

ld-xy avatar Sep 27 '22 09:09 ld-xy

先转换为numpy.array, 然后torch.from_numpy

lyyiangang avatar Oct 02 '22 15:10 lyyiangang

请给出具体的复现方法

wangzhaode avatar Nov 01 '22 03:11 wangzhaode

macos平台:使用下面这两行代码,得到的类型是MNN.tensor类型,这种类型能转化成torch.tensor然后接着执行模型下面的代码吗?

    tmp_output = MNN.Tensor((1, 240, 384), MNN.Halide_Type_Float, np.ones([1, 240, 384]).astype(np.float32),
                            MNN.Tensor_DimensionType_Caffe)
    output_tensor.copyToHostTensor(tmp_output)

如果可以的话,怎么转换啊,求助!!!

np_data = output_tensor.read() # convert to numpy.ndarray
torch_data = torch. from_numpy(np_data) # convert ndarray to torch.Tensor

wangzhaode avatar Nov 01 '22 03:11 wangzhaode