Toybrick

rknn推理时输入不同图片,推理结果一样

toybricklxq

新手上路

积分
30
楼主
发表于 2020-12-4 18:06:24    查看: 5975|回复: 0 | [复制链接]    打印 | 只看该作者
目前使用rknn3399pro在开发板上使用作推理,运行多次不同输入数据,其推理得到的结果都是一样的。
模型推理主要代码如下:

模型转换:
rknn = RKNN()
rknn.load_darknet(model = cfg ,  weight=weights)
rknn.config(channel_mean_value='0 0 0 255',target_platform=['rk3399pro'], reorder_channel='0 1 2', batch_size=1)
rknn.build(do_quantization=False, dataset='./dataset_608x608.txt', pre_compile=False)
rknn.export_rknn('./yolov3_change.rknn')
rknn.release()


推理:
rknn = RKNN()
rknn.load_rknn('./yolov3_608x608.rknn')
ret = rknn.init_runtime()

pred = rknn.inference(inputs=[mat])

在测试时做过多次print推理结果后就exit()终止程序。
麻烦各位遇到过该问题的人帮帮忙!!!
回复

使用道具 举报

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

本版积分规则

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


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