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 (http://t.rock-chips.com/) Powered by Discuz! X3.3