Toybrick

标题: rknn推理时输入不同图片，推理结果一样 [打印本页]

作者: toybricklxq 时间: 2020-12-4 18:06
标题: rknn推理时输入不同图片，推理结果一样
目前使用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()终止程序。
麻烦各位遇到过该问题的人帮帮忙！！！

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