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

[复制链接] · 发表于 2020-9-27 09:37:26

rknn_init初始化加载YOLOV3检测模型后，有无方法得到目标类别数？在其他芯片上(例如海思)，可以加载模型后直接得到YOLOV3检测的类别数，这样的话可加载不同模型自动刷新类别数，而不是现在这样需要提前告知有多少类别，便于动态基于模型得到类别信息

只看该作者 · 发表于 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 属性。