View the Model Information

The hrt_model_exec model_info tool can be used to obtain the model information.

Support Range

The support scope of the model: QAT(Quantized Awareness Training) model, PTQ (Post-Training Quantization) model.

Usage

Usage:
hrt_model_exec [Option...] [Parameter]

[Option]                      [instruction]
---------------------------------------------------------------------------------------------------------------
-h --help                     Display this information
-v --version                  Display this version

[Option]                      [Parameter]
---------------------------------------------------------------------------------------------------------------
--model_file                  [string]: Model file paths, separate by comma, each represents one model file path.
--model_name                  [string]: Model name.
                                        When model_file has one more model and Subcommand is infer or perf,
                                        "model_name" must be specified!                                      
[Examples]
--------------------------------------------------------------------------------------------------------------------
hrt_model_exec model_info     
   --model_file       
   --model_name            

Parameters Introduction

ParameterData TypeParameter DescriptionCorrelated Parameters
-h, --helpNone.Display the help information.None.
-v, --versionNone.View the version number of the DNN prediction library of the tool.None.
model_infoNone.Get model information, such as model input and output information, etc.This parameter is used together with model_file to get detailed information about the model, including model input and output information hbDNNTensorProperties.
model_filestringModel file path, multiple paths can be separated by commas.None.
model_namestringSpecify the name of a model.None.

Usage Example

Obtain the information of a single model

hrt_model_exec model_info --model_file=xxx.hbm

../aarch64/bin/hrt_model_exec model_info --model_file=resnet50_224x224_nv12.hbm

Load model to DDR cost 1965.57ms.
This model file has 1 model:
[resnet50_224x224_nv12]
---------------------------------------------------------------------
[model name]: resnet50_224x224_nv12

input[0]:
name: input_y
valid shape: (1,224,224,1,)
aligned byte size: -1
tensor type: HB_DNN_TENSOR_TYPE_U8
quanti type: NONE
stride: (-1,-1,1,1,)

input[1]:
name: input_uv
valid shape: (1,112,112,2,)
aligned byte size: -1
tensor type: HB_DNN_TENSOR_TYPE_U8
quanti type: NONE
stride: (-1,-1,2,1,)

output[0]:
name: output
valid shape: (1,1000,)
aligned byte size: 4096
tensor type: HB_DNN_TENSOR_TYPE_F32
quanti type: NONE
stride: (4000,4,)

---------------------------------------------------------------------

Obtain the information of multiple models

  1. Output the information of all models.
hrt_model_exec model_info --model_file=xxx.hbm,xxx.hbm  
  1. Output the information of the specified pack model.
hrt_model_exec model_info --model_file=xxx.hbm --model_name=xx