Toybrick

标题: rknn_init初始化加载YOLOV3检测模型后,可得到类别数? [打印本页]

作者: whx_huxin    时间: 2020-9-27 09:37
标题: rknn_init初始化加载YOLOV3检测模型后,可得到类别数?
rknn_init初始化加载YOLOV3检测模型后,有无方法得到目标类别数?在其他芯片上(例如海思),可以加载模型后直接得到YOLOV3检测的类别数,这样的话可加载不同模型自动刷新类别数,而不是现在这样需要提前告知有多少类别,便于动态基于模型得到类别信息


作者: brunolin    时间: 2020-10-5 10:21
可以用rknn_query查詢tensor數據元素個數,再推算類別數
RKNN_QUERY_IN_OUT_NUM  查询input 和output 的Tensor 个数。
RKNN_QUERY_INPUT_ATTR  查询Input Tensor 属性。
RKNN_QUERY_OUTPUT_ATTR  查询Output Tensor 属性。




欢迎光临 Toybrick (http://t.rock-chips.com/) Powered by Discuz! X3.3