hb_model_info 是用于解析*.hbm和*.bc编译时的依赖及参数信息、*.onnx模型基本信息,同时支持对*.bc可删除节点进行查询的工具。
hb_model_info ${model_file}
| 参数名称 | 参数说明 |
|---|---|
-h, --help | 显示帮助信息。 |
--version | 显示版本并退出。 |
-n, --name | 后接模型名称。在HBM_FILE为pack模型时指定所需模型,用于输出该指定模型的模型编译信息。 |
-v, --visualize | 启动webserver以查看模型结构,查看完毕后可使用 Ctrl+C 关闭。 |
输出部分将会是模型编译时的一些输入信息,如下所示:
以下代码块中的版本号信息等内容将随发布包版本变化,此处仅为示例。
log will be stored in /open_explorer/hb_model_info.log
Start hb_model_info....
hb_model_info version 3.0.8
hbm_path: /open_explorer/mobilenetv1_224x224_nv12.hbm
desc file path: mobilenetv1_224x224_nv12_desc.json
************* mobilenetv1_224x224_nv12 *************
############# model deps info #############
builder version : 3.0.8
hbdk version : 4.0.17
horizon nn version : 1.0.3
############# model_parameters info #############
caffe_model : /open_explorer/samples/ai_toolchain/horizon_model_convert_sample/01_common/model_zoo/mapper/classification/mobilenet/mobilenet.caffemodel
prototxt : /open_explorer/samples/ai_toolchain/horizon_model_convert_sample/01_common/model_zoo/mapper/classification/mobilenet/mobilenet_deploy.prototxt
BPU march : nash-e
layer_out_dump : False
working dir : /open_explorer/samples/ai_toolchain/horizon_model_convert_sample/03_classification/01_mobilenetv1/model_output
output_model_file_prefix: mobilenetv1_224x224_nv12
node_info : {}
############# input_parameters info #############
------------------------------------------
---------input info : data ---------
input_name : data
input_type_rt : nv12
input_space&range : regular
input_type_train : bgr
input_layout_train : NCHW
norm_type : data_mean_and_scale
input_shape : 1x3x224x224
mean_value : [103.94, 116.78, 123.68]
scale_value : [0.017]
separate_batch : False
---------input info : data end -------
------------------------------------------
############# calibration_parameters info #############
calibration_type : max
cal_data_dir : /open_explorer/samples/ai_toolchain/horizon_model_convert_sample/03_classification/01_mobilenetv1/calibration_data_bgr
max_percentile : 0.9999
per_channel : False
############# compiler_parameters info #############
debug : True
optimize_level : O2
compile_mode : latency
core_num : 1
balance_factor : 100
input_source : {'data': 'pyramid'}
hbm_path: /open_explorer/mobilenetv1_224x224_nv12.hbm
############# Model input/output info #############
NAME TYPE SHAPE DATA_TYPE
---- ------ ---------------- ---------
data input [1, 224, 224, 3] uint8
prob output (1, 1000, 1, 1) float32