Vision SDK Python接口中,怎么将Tensor转换成numpy
收藏回复举报
Vision SDK Python接口中,怎么将Tensor转换成numpy
新人帖
发表于2025-11-01 15:19:11
0 查看

在推理完之后,将Tensor转化到了host端,怎么继续将Tensor转成numpy,我想使用numpy格式数据上实现自定义的后处理,以下是我的推理:

def predict(self, input_img):
        img_tensor = Tensor(input_img)
        outputs = self.net.infer(img_tensor)
        output = outputs[0]
        output.to_host()
        return output

我想问,怎么把这个output转成numpy,我查了相关python API,怎么没有相关的接口?

我要发帖子