9.5.1. hrt_bin_dump工具介绍

9.5.1.1. 工具简介

hrt_bin_dump 是模型的layer dump工具,工具的输出文件为二进制文件。

9.5.1.2. 输入参数描述

编号

参数

类型

描述

说明

1

model_file

string

模型文件路径。

指定模型文件路径,可dump模型所有节点的输入和输出文件

2

input_file

string

输入文件路径。

模型的输入文件,支持 hbDNNDataType 所有类型的输入;

IMG类型文件需为二进制文件(后缀必须为.bin),二进制文件的大小应与模型的输入信息相匹配, 如:YUV444文件大小为 \(height * width * 3\)

TENSOR类型文件需为二进制文件或文本文件(后缀必须为.bin/.txt),二进制文件的大小应与模型的输入信息相匹配, 文本文件的读入数据个数必须大于等于模型要求的输入数据个数,多余的数据会被丢弃;

每个输入之间通过逗号分隔,如:模型有两个输入,则: --input_file=kite.bin,input.txt

3

dump_path

string

工具输出路径。

工具的输出路径,该路径应为合法路径。

9.5.1.3. 使用说明

工具提供模型节点输入输出dump功能,输出文件为二进制文件。 直接运行 hrt_bin_dump 获取工具使用详情。 参见下图:

../../../_images/hrt_bin_dump_help.png

工具可以通过 -v 或者 --version 命令,查看工具的 dnn 预测库版本号。

hrt_bin_dump -v
hrt_bin_dump --version

9.5.1.3.1. 示例

以mobilenetv1模型为例,创建outputs文件夹,执行以下命令:

./hrt_bin_dump --model_file=./mobilenetv1.bin --dump_path=./outputs --input_file=./input.bin

运行日志参见以下截图:

../../../_images/run_log.png

在路径 outputs/ 文件夹下可以查看输出,参见以下截图:

../../../_images/output.png