请教，rknn的输出如何保存图像？

[复制链接] · 发表于 2019-6-12 22:29:52

本帖最后由 alonging 于 2019-6-12 22:29 编辑

如下代码，rkkn输出的图像保存失败，哪边的问题？应该如何操作的？ outputs = rknn.inference(inputs=[orig_img])
mat = np.array(outputs)
img = mat.reshape((1,3,256,256))
mypython_umat = cv2.UMat(img)
cv2.imwrite('out.jpg', mypython_umat)

显示全部楼层 · 发表于 2019-6-13 12:44:59

troy 发表于 2019-6-13 08:43
你的模型输出是一张图像吗？

是一张图，啥问题呢，版主帮忙哈