hb_model_info
是用于解析*.hbm和*.bc编译时的依赖及参数信息、*.onnx模型基本信息,同时支持对*.bc可删除节点进行查询的工具,可通过启动webserver以查看模型结构。
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/samples/ai_toolchain/horizon_model_convert_sample/03_classification/03_resnet50/hb_model_info.log
Start hb_model_info....
hb_model_info version 3.3.0
hbm_path: /open_explorer/samples/ai_toolchain/horizon_model_convert_sample/03_classification/03_resnet50/model_output/resnet50_224x224_nv12.hbm
desc file path: resnet50_224x224_nv12_desc.json
************* resnet50_224x224_nv12 *************
############# model deps info #############
builder version : 3.3.0
hbdk version : 4.1.2
horizon nn version : 2.0.5
############# model_parameters info #############
onnx_model : /open_explorer/samples/ai_toolchain/horizon_model_convert_sample/01_common/model_zoo/mapper/classification/resnet50/resnet50.onnx
BPU march : nash-e
layer_out_dump : False
working dir : /open_explorer/samples/ai_toolchain/horizon_model_convert_sample/03_classification/03_resnet50/model_output
output_model_file_prefix: resnet50_224x224_nv12
node_info : {}
############# input_parameters info #############
------------------------------------------
---------input info : input ---------
input_name : input
input_type_rt : nv12
input_space&range : regular
input_type_train : rgb
input_layout_train : NCHW
input_shape : 1x3x224x224
mean_value : [123.675, 116.28, 103.53]
scale_value : [0.01712475, 0.017507, 0.01742919]
separate_batch : False
---------input info : input end -------
------------------------------------------
############# calibration_parameters info #############
calibration_type : default
max_percentile : None
per_channel : False
cal_data_dir : /open_explorer/samples/ai_toolchain/horizon_model_convert_sample/03_classification/03_resnet50/calibration_data_rgb
############# compiler_parameters info #############
debug : True
optimize_level : O2
compile_mode : latency
core_num : 1
balance_factor : 100
input_source : {'input': 'pyramid'}
hbm_path: /open_explorer/samples/ai_toolchain/horizon_model_convert_sample/03_classification/03_resnet50/model_output/resnet50_224x224_nv12.hbm
############# Model input/output info #############
NAME TYPE SHAPE DATA_TYPE
------ ------ ---------------- ---------
input_y input [1, 224, 224, 1] UINT8
input_uv input [1, 112, 112, 2] UINT8
output output [1, 1000] FLOAT32