Toybrick

rknn_init初始化加载YOLOV3检测模型后,可得到类别数?

whx_huxin

注册会员

积分
97
发表于 2020-9-27 09:37:26    查看: 803|回复: 1 | [复制链接]    打印 | 显示全部楼层
rknn_init初始化加载YOLOV3检测模型后,有无方法得到目标类别数?在其他芯片上(例如海思),可以加载模型后直接得到YOLOV3检测的类别数,这样的话可加载不同模型自动刷新类别数,而不是现在这样需要提前告知有多少类别,便于动态基于模型得到类别信息

回复

使用道具 举报

brunolin

中级会员

积分
441
发表于 2020-10-5 10:21:24 | 显示全部楼层
可以用rknn_query查詢tensor數據元素個數,再推算類別數
RKNN_QUERY_IN_OUT_NUM  查询input 和output 的Tensor 个数。
RKNN_QUERY_INPUT_ATTR  查询Input Tensor 属性。
RKNN_QUERY_OUTPUT_ATTR  查询Output Tensor 属性。
回复

使用道具 举报

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

本版积分规则

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


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