Toybrick

请教,rknn的输出如何保存图像?

alonging

注册会员

积分
55
发表于 2019-6-12 22:29:52    查看: 9843|回复: 3 | [复制链接]    打印 | 显示全部楼层
本帖最后由 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

版主

积分
2376
发表于 2019-6-13 08:43:07 | 显示全部楼层
你的模型输出是一张图像吗?
回复

使用道具 举报

alonging

注册会员

积分
55
 楼主| 发表于 2019-6-13 12:44:59 | 显示全部楼层
troy 发表于 2019-6-13 08:43
你的模型输出是一张图像吗?

是一张图,啥问题呢,版主帮忙哈
回复

使用道具 举报

troy

版主

积分
2376
发表于 2019-6-26 15:30:41 | 显示全部楼层
alonging 发表于 2019-6-13 12:44
是一张图,啥问题呢,版主帮忙哈

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

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表