D-Robotics Official Website
D-Robotics Developer Community
D-Robotics Official Website
D-Robotics Developer Community
English

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