Toybrick

标题: 请教，rknn的输出如何保存图像？ [打印本页]

作者: alonging 时间: 2019-6-12 22:29
标题: 请教，rknn的输出如何保存图像？
本帖最后由 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)

作者: troy 时间: 2019-6-13 08:43
你的模型输出是一张图像吗？

作者: alonging 时间: 2019-6-13 12:44

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

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

作者: troy 时间: 2019-6-26 15:30

alonging 发表于 2019-6-13 12:44
是一张图，啥问题呢，版主帮忙哈

形状reshape一下，interface的输出一般一个list，然后里面的成员是ndarray的数组，你可以把内容reshape成图片的形状，然后用opencv的函数，比如cv2.imwrite("test.jpg", data)就可以保存了。

欢迎光临 Toybrick (https://t.rock-chips.com/)