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 (https://t.rock-chips.com/)