hb_model_info工具

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