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

    如下代码，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
是一张图，啥问题呢，版主帮忙哈

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