Page 2 of 2

Re: 输入非图片的神经网络模型转rknn

Posted: 2025-09-29 1:49
by tangj
beamrobot wrote: 2024-07-28 2:29 我查看手册发现校验数据dataset也支持npy格式,所以我在导出模型时把输入“包装”成了图片的NCHW格式。这样可以正常的转换成rknn格式了,python中验证也基本正确了。但是现在的问题是,在板子上用C程序跑模型,输出与python版本不一致(都是错的),请问这是什么原因?C代码我参考了retinaface例子的inference_retinaface_model函数,它是像下面这样拿输出结果的,我照葫芦画瓢把我想要的向量从这里取出来,这样做对吗?
uint8_t *location = (uint8_t *)(app_ctx->output_mems[0]->virt_addr);
请问后面你成功了吗?最后模型结果正确吗?