华为计算微信公众号
昇腾AI开发者公众号
华为计算微博
华为计算今日头条
在推理完之后,将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,怎么没有相关的接口?
我要发帖子
在推理完之后,将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,怎么没有相关的接口?