4.5.4.2. 评测方法说明

简介

本章节介绍公版模型精度性能评测 ai_benchmark 示例包的具体用法， 示例包中预置了源码、可执行程序和评测脚本，开发者可以直接在地平线开发板上体验并基于这些示例进行应用开发，降低开发门槛。

示例包提供常见的分类、检测和分割模型的性能评测和精度评测示例，详细内容请阅读下文。

小技巧

公版模型精度性能评测 ai_benchmark 示例包获取，请参考 《交付物说明》。

示例代码包结构

+---ai_benchmark
| +---code                           # 示例源码文件夹
| | +---build_ptq_xj3.sh
| | +---CMakeLists.txt
| | +---deps/deps_gcc9.3             # 第三方依赖库，gcc6.5环境为deps，gcc9.3为deps_gcc9.3
|   | +---aarch64
| | +---include                      # 源码头文件
|   | +---base
|   | +---input
|   | +---method
|   | +---output
|   | +---plugin
|   | +---utils
| | +---src                          # 示例源码
|   | +---input
|   | +---method
|   | +---output
|   | +---plugin
|   | +---utils
|   | +---simple_example.cc          # 示例主程序
| +---xj3                             # 示例包运行环境
|   +---ptq                          # 后量化模型示例
|   | +---data                       # 模型精度评测数据集
|   | +---mini_data                  # 模型性能评测数据集
|   | +---model                      # 后量化nv12模型
|   | +---script                     # 执行脚本
|   | | +---aarch64                  # 编译产生可执行文件及依赖库
|   | | +---base_config.sh
|   | | +---config
|   | | +---classification           # 分类模型示例
|   | | | +---efficientnet_lite0
|   | | | | +---accuracy.sh          # 模型精度示例脚本
|   | | | | +---fps.sh               # 模型性能示例脚本
|   | | | | +---latency.sh           # 模型单帧延时示例脚本
|   | | | | +---workflow_accuracy.json
|   | | | | +---workflow_fps.json
|   | | | | +---workflow_latency.json
|   | | | +---efficientnet_lite1
|   | | | +---efficientnet_lite2
|   | | | +---efficientnet_lite3
|   | | | +---efficientnet_lite4
|   | | | +---googlenet
|   | | | +---mobilenetv1
|   | | | +---mobilenetv2
|   | | | +---resnet18
|   | | +---detection                # 检测模型示例
|   | | | +---efficientdetd0
|   | | | +---centernet_resnet50
|   | | | +---fcos_efficientnetb0
|   | | | +---ssd_mobilenetv1
|   | | | +---yolov2_darknet19
|   | | | +---yolov3_darknet53
|   | | | +---yolov5s
|   | | +---segmentation             # 分割模型示例
|   | | | +---unet_mobilenet
|   | +---tools                      # 精度评测工具
|   | | +---python_tools
|   | | | +---accuracy_tools

• code：该目录内是评测程序的源码，用来进行模型性能和精度评测。

• xj3： 提供了已经编译好的应用程序，以及各种评测脚本，用来测试多种模型在地平线BPU上运行的性能，精度等。

• build_ptq_xj3.sh：开发板程序一键编译脚本。

• deps/deps_gcc9.3：示例代码所需要的依赖，主要如下所示:

gflags  glog  hobotlog nlohmann opencv  rapidjson

示例模型

我们提供了开源的模型库，里面包含常用的分类、检测和分割模型，模型的命名规则为：{model_name}_{backbone}_{input_size}_{input_type}，开发者可以直接使用。

小技巧

以下表格中的bin模型都是通过 horizon_model_convert_sample 模型转换示例包转换编译出来的，模型存放在 model_zoo 文件夹中，请阅读 《交付物说明》 章节内容进行获取。

MODEL	MODEL NAME
centernet_resnet50	centernet_resnet50_512x512_nv12.bin
efficientdetd0	efficientdetd0_512x512_nv12.bin
efficientnet_lite0	efficientnet_lite0_224x224_nv12.bin
efficientnet_lite1	efficientnet_lite1_240x240_nv12.bin
efficientnet_lite2	efficientnet_lite1_260x260_nv12.bin
efficientnet_lite3	efficientnet_lite1_280x280_nv12.bin
efficientnet_lite4	efficientnet_lite1_300x300_nv12.bin
fcos_efficientnetb0	fcos_efficientnetb0_512x512_nv12.bin
googlenet	googlenet_224x224_nv12.bin
googlenet_cop	googlenet_cop_224x224_nv12.bin.bin
lenet_gray	lenet_28x28_gray.bin
mobilenet_multi	mobilenet_multi_224x224_gray.bin
ssd_mobilenetv1	ssd_mobilenetv1_300x300_nv12.bin
unet_mobilenet	unet_mobilenet_1024x2048_nv12.bin
mobilenetv1	mobilenetv1_224x224_nv12.bin
	mobilenetv1_224x224_nv12_dump.bin
	mobilenetv1_128x128_resizer_nv12.bin
mobilenetv2	mobilenetv2_224x224_nv12.bin
resnet18	resnet18_224x224_nv12.bin
resnet50_feature	resnet50_64x56x56_featuremap.bin
yolov2_darknet19	yolov2_darknet19_608x608_nv12.bin
	yolov2_darknet19_preempted_608x608_nv12.bin
yolov3_darknet53	yolov3_darknet53_416x416_nv12.bin
	yolov3_darknet53_preempted_416x416_nv12.bin
yolov5s	yolov5s_672x672_nv12.bin

公共数据集

测评示例中用到的数据集主要有VOC数据集、COCO数据集、ImageNet数据集和Cityscapes数据集。

请在linux环境下进行下载，获取方式如下：

VOC：wget -c ftp://vrftp.horizon.ai/Open_Explorer/eval_dataset/VOC.tar.gz
coco：wget -c ftp://vrftp.horizon.ai/Open_Explorer/eval_dataset/coco.tar.gz
imagenet：wget -c ftp://vrftp.horizon.ai/Open_Explorer/eval_dataset/imagenet.tar.gz
cityscapes：wget -c ftp://vrftp.horizon.ai/Open_Explorer/eval_dataset/cityscapes.tar.gz

环境构建

开发板准备

1. 拿到开发板后，请按照镜像升级说明升级系统到地平线整包交付物SDK对应的系统版本。开发板升级步骤，请阅读 镜像升级方法 章节。

2. 确保本地开发机和开发板可以远程连接。

编译环境准备

​编译需要当前环境安装好交叉编译工具 gcc-ubuntu-9.3.0-2020.03-x86_64-aarch64-linux-gnu。若使用地平线提供的开发机模型转换编译环境，请直接进行编译使用。开发机模型转换编译环境的使用方法，请阅读 环境安装 章节内容， 反之，请检查gcc交叉编译器是否正确安装，请阅读 安装交叉编译工具链 章节内容。 请使用code目录下的 build_ptq_xj3.sh 脚本，即可一键编译开发板环境下的可执行程序，可执行程序和对应依赖会自动复制到 xj3/ptq/script 目录下的 aarch64 目录下。

注解

需要注意build_ptq_xj3.sh脚本里指定的交叉编译工具链的位置是 /opt 目录下，用户如果安装在其他位置，可以手动修改下build_ptq_xj3.sh。

export CC=/opt/gcc-ubuntu-9.3.0-2020.03-x86_64-aarch64-linux-gnu/bin/aarch64-linux-gnu-gcc
export CXX=/opt/gcc-ubuntu-9.3.0-2020.03-x86_64-aarch64-linux-gnu/bin/aarch64-linux-gnu-g++

测评示例使用说明

​评测示例脚本主要在 script 和 tools 目录下。 script是开发板上运行的评测脚本，包括常见分类，检测和分割模型。每个模型下面有三个脚本，分别表示：

• fps.sh利用多线程调度实现fps统计，用户可以根据需求自由设置线程数。

• latency.sh实现单帧延迟统计（一个线程，单帧）。

• accuracy.sh用于精度评测。

script:

├── aarch64             # 编译产生的可执行文件及依赖库
│   ├── bin
│   ├── lib
├── base_config.sh      # 基础配置
├── config              # image_name配置文件
│   ├── data_name_list
|   |   ├── coco_detlist.list
│   |   ├── imagenet.list
│   |   ├── voc_detlist.list
├── classification      # 分类模型评测
│   ├── efficientnet_lite0
│   │   ├── accuracy.sh
│   │   ├── fps.sh
│   │   ├── latency.sh
│   │   ├── workflow_accuracy.json
│   │   ├── workflow_fps.json
│   │   ├── workflow_latency.json
│   ├── mobilenetv1
│   ├── .....
│   ├── resnet18
├── detection           # 检测模型
|   ├── centernet_resnet50
│   │   ├── accuracy.sh
│   │   ├── fps.sh
│   │   ├── latency.sh
│   │   ├── workflow_accuracy.json
│   │   ├── workflow_fps.json
│   │   ├── workflow_latency.json
│   ├── yolov2_darknet19
│   ├── yolov3_darknet53
│   ├── ...
│   ├── efficientdetd0
└── segmentation       # 分割模型
    └──unet_mobilenet
        ├── accuracy.sh
        ├── fps.sh
        ├── latency.sh
        ├── workflow_accuracy.json
        ├── workflow_fps.json
        ├── workflow_latency.json

tools目录下是精度评测需要的脚本。主要包括 python_tools 下的精度计算脚本。

tools:

python_tools
  └── accuracy_tools

注意

评测前需要执行以下命令，将 ptq 目录拷贝到开发板上，然后将 model_zoo/runtime 拷贝到 ptq/model 目录下。

scp -r ai_toolchain_package/Ai_Toolchain_Package-release-v1.16.4-OE-v2.2.3a/ai_benchmark/xj3/ptq root@192.168.1.10:/userdata/ptq/

scp -r model_zoo/runtime root@192.168.1.10:/userdata/ptq/model/

性能评测

性能评测分为latency和fps两方面。

测评脚本使用说明

进入到需要评测的模型目录下，执行 sh latency.sh 即可测试出单帧延迟。如下图所示：

I0419 02:35:07.041095 39124 output_plugin.cc:80]  Infer latency:  [avg:  13.124ms,  max:  13.946ms,  min:  13.048ms], Post process latency: [avg:  3.584ms,  max:  3.650ms,  min:  3.498ms].

注解

• infer 表示模型推理耗时。

• Post process 表示后处理耗时。

进入到需要评测的模型目录下执行 sh fps.sh 即可测试出帧率。如下图所示：

I0419 02:35:00.044417 39094 output_plugin.cc:109]  Throughput: 176.39fps      # 模型帧率

注解

该功能采用多线程并发方式，旨在让模型可以在BPU上达到极致的性能。由于多线程并发及数据采样的原因，在程序启动阶段帧率值会较低，之后帧率会上升并逐渐趋于稳定，帧率的浮动范围控制在0.5%之内。

命令行参数说明

accuracy.sh 脚本内容如下：

#!/bin/sh

source ../../base_config.sh                      # 加载基础配置
export SHOW_FPS_LOG=1                            # 设置环境变量，打印fps级别log

${app} \                                         # 可执行程序，在accuracy.sh脚本中定义
  --config_file=workflow_accuracy.json \         # 加载精度测试workflow配置文件
  --log_level=2                                  # 设置log等级

fps.sh 脚本内容如下：

#!/bin/sh

source ../../base_config.sh
export SHOW_FPS_LOG=1
export STAT_CYCLE=10                             # 设置环境变量，FPS 统计周期

${app} \
  --config_file=workflow_fps.json \
  --log_level=1

latency.sh 脚本内容如下：

#!/bin/sh

source ../../base_config.sh
export SHOW_LATENCY_LOG=1                            # 设置环境变量，打印 LATENCY 级别log
export STAT_CYCLE=5                                  # 设置环境变量，LATENCY 统计周期

${app} \
  --config_file=workflow_latency.json \
  --log_level=1

配置文件说明

注解

注意：max_cache参数生效时会预处理图片并读取到内存中，为保障您的程序稳定运行，请不要设置过大的值，建议您的数值设置不超过30。

以fcos_efficientnetb0模型为例，workflow_accuray.json 配置文件内容如下：

{
  "input_config": {
    "input_type": "preprocessed_image",
    "height": 512,
    "width": 512,
    "data_type": 1,
    "image_list_file": "../../../data/coco/coco.lst",
    "need_pre_load": false,
    "need_loop": false,
    "max_cache": 10
  },
  "output_config": {
    "output_type": "eval",
    "eval_enable": true,
    "output_file": "./eval.log"
  },
  "workflow": [
    {
      "method_type": "InferMethod",
      "unique_name": "InferMethod",
      "method_config": {
        "core": 0,
        "model_file": "../../../model/runtime/fcos_efficientnetb0/fcos_efficientnetb0_512x512_nv12.bin"
      }
    },
    {
      "thread_count": 2,
      "method_type": "PTQFcosPostProcessMethod",
      "unique_name": "PTQFcosPostProcessMethod",
      "method_config": {
        "strides": [
          8,
          16,
          32,
          64,
          128
        ],
        "class_num": 80,
        "score_threshold": 0.05,
        "topk": 1000,
        "det_name_list": "../../config/data_name_list/coco_detlist.list"
      }
    }
  ]
}

workflow_fps.json 配置文件内容如下：

{
  "input_config": {
    "input_type": "image",
    "height": 512,
    "width": 512,
    "data_type": 1,
    "image_list_file": "../../../mini_data/coco/coco.lst",
    "need_pre_load": true,
    "limit": 10,
    "need_loop": true,
    "max_cache": 10
  },
  "output_config": {
    "output_type": "image",
    "image_list_enable": true,
    "image_output_dir": "./output_images",
    "in_order": false
  },
  "workflow": [
    {
      "method_type": "InferMethod",
      "unique_name": "InferMethod",
      "method_config": {
        "core": 0,
        "model_file": "../../../model/runtime/fcos_efficientnetb0/fcos_efficientnetb0_512x512_nv12.bin"
      }
    },
    {
      "thread_count": 4,
      "method_type": "PTQFcosPostProcessMethod",
      "unique_name": "PTQFcosPostProcessMethod",
      "method_config": {
        "strides": [
          8,
          16,
          32,
          64,
          128
        ],
        "class_num": 80,
        "score_threshold": 0.5,
        "topk": 1000,
        "det_name_list": "../../config/data_name_list/coco_detlist.list"
      }
    }
  ]
}

workflow_latency.json 如下：

{
  "input_config": {
    "input_type": "image",
    "height": 512,
    "width": 512,
    "data_type": 1,
    "image_list_file": "../../../mini_data/coco/coco.lst",
    "need_pre_load": true,
    "limit": 1,
    "need_loop": true,
    "max_cache": 10
  },
  "output_config": {
    "output_type": "image",
    "image_list_enable": true,
    "image_output_dir": "./output_images"
  },
  "workflow": [
    {
      "method_type": "InferMethod",
      "unique_name": "InferMethod",
      "method_config": {
        "core": 0,
        "model_file": "../../../model/runtime/fcos_efficientnetb0/fcos_efficientnetb0_512x512_nv12.bin"
      }
    },
    {
      "thread_count": 1,
      "method_type": "PTQFcosPostProcessMethod",
      "unique_name": "PTQFcosPostProcessMethod",
      "method_config": {
        "strides": [
          8,
          16,
          32,
          64,
          128
        ],
        "class_num": 80,
        "score_threshold": 0.5,
        "topk": 1000,
        "det_name_list": "../../config/data_name_list/coco_detlist.list"
      }
    }
  ]
}

精度评测

​模型精度评测分为四步：

1. 数据预处理。

2. 数据挂载。

3. 模型推理。

4. 精度计算。

数据预处理

​对于PTQ模型：数据预处理需要在x86开发机环境下运行 hb_eval_preprocess 工具，对数据集进行预处理。 所谓预处理是指图片数据在送入模型之前的特定处理操作，例如：图片resize、crop和padding等。 该工具集成于开发机模型转换编译的环境中，原始数据集经过工具预处理之后，会生成模型对应的前处理二进制文件.bin文件集. 直接运行 hb_eval_preprocess --help 可查看工具使用规则。

小技巧

1. 关于 hb_eval_preprocess 工具命令行参数，可键入 hb_eval_preprocess -h， 或查看 PTQ量化原理及步骤说明的 hb_eval_preprocess工具 一节内容。

下面将详细介绍示例包中每一个模型对应的数据集，以及对应数据集的预处理操作：

• VOC数据集 ：该数据集主要用于ssd_mobilenetv1模型的评测， 其目录结构如下，示例中主要用到 Main 文件下的val.txt文件， JPEGImages 中的源图片和 Annotations 中的标注数据：

.
└── VOCdevkit                  # 根目录
    └── VOC2012                # 不同年份的数据集，这里只下载了2012的，还有2007等其它年份的
        ├── Annotations        # 存放xml文件，与JPEGImages中的图片一一对应，解释图片的内容等等
        ├── ImageSets          # 该目录下存放的都是txt文件，txt文件中每一行包含一个图片的名称，末尾会加上±1表示正负样本
        │   ├── Action
        │   ├── Layout
        │   ├── Main
        │   └── Segmentation
        ├── JPEGImages         # 存放源图片
        ├── SegmentationClass  # 存放的是图片，语义分割相关
        └── SegmentationObject # 存放的是图片，实例分割相关

对数据集进行预处理：

hb_eval_preprocess -m ssd_mobilenetv1 -i VOCdevkit/VOC2012/JPEGImages -v VOCdevkit/VOC2012/ImageSets/Main/val.txt -o ./pre_ssd_mobilenetv1

• COCO数据集 ：该数据集主要用于yolov2_darknet19、yolov3_darknet53、yolov5s、efficientdetd0、fcos_efficientnetb0和centernet_resnet50模型的评测， 其目录如下，示例中主要用到 annotations 文件夹下的instances_val2017.json标注文件和 images 中的图片：

.
├── annotations    # 存放标注数据
└── images         # 存放源图片

对数据集进行预处理：

hb_eval_preprocess -m model_name -i coco/coco_val2017/images -o ./pre_model_name

• ImageNet数据集 ：该数据集主要用于EfficientNet_lite0、EfficientNet_Lite1、EfficientNet_Lite2、EfficientNet_Lite3、EfficientNet_Lite4、MobileNet、GoogleNet、ResNet等分类模型的评测， 示例中主要用到了标注文件val.txt 和 val 目录中的源图片:

.
├── val.txt
└── val

对数据集进行预处理：

hb_eval_preprocess -m model_name -i imagenet/val -o ./pre_model_name

• Cityscapes数据集 ：该数据集用于unet_mobilenet模型的评测。 示例中主要用到了 ./gtFine/val 中的标注文件和 ./leftImg8bit/val 中的源图片。

.
├── gtFine
│   └── val
│       ├── frankfurt
│       ├── lindau
│       └── munster
└── leftImg8bit
    └── val
        ├── frankfurt
        ├── lindau
        └── munster

对数据集进行预处理：

hb_eval_preprocess -m unet_mobilenet -i cityscapes/leftImg8bit/val -o ./pre_unet_mobilenet

示例中精度计算脚本的运行流程是：

1、根据 workflow_accurary.json 中的 image_list_file 参数值，去寻找对应数据集的 lst 文件；

2、根据 lst 文件存储的前处理文件路径信息，去加载每一个前处理文件，然后进行推理

所以，生成预处理文件之后，需要生成对应的lst文件，将每一张前处理文件的路径写入到lst文件中，而这个路径与数据集在开发板端的存放位置有关。 这里我们推荐其存放位置与 script 文件夹同级目录，如下：

|-- ptq
|   |-- data
|   |   |-- cityscapes
|   |   |   -- xxxx.bin             # 前处理好的二进制文件
|   |   |   -- ....
|   |   |   -- cityscapes.lst       # lst文件：记录每一个前处理文件的路径
|   |   |-- coco
|   |   |   -- xxxx.bin
|   |   |   -- ....
|   |   |   -- coco.lst
|   |   |-- imagenet
|   |   |   -- xxxx.bin
|   |   |   -- ....
|   |   |   -- imagenet.lst
|   |   `-- voc
|   |   |   -- xxxx.bin
|   |   |   -- ....
|   |       `-- voc.lst
|   |-- model
|   |   |-- ...
|   |-- script
|   |   |-- ...

与之对应的lst文件，参考生成方式如下：

find ../../../data/coco/fcos -name "*bin*" > ../../../data/coco/coco.lst

这样生成的lst文件中存储的路径为一个相对路径： ../../../data/ ，可以与 workflow_accurary.json 默认的配置路径吻合。 如果需要更改前处理数据集的存放位置，则需要确保对应的 lst 文件可以被 workflow_accurary.json 读取到；其次需要确保程序根据 lst 中的路径信息，能读取到对应的前处理文件。

数据挂载

由于数据集相对较大，不适合直接放在开发板上，可以采用nfs挂载的方式供开发板读取。

开发机PC端（需要root权限）：

1. 编辑 /etc/exports, 增加一行： /nfs *(insecure,rw,sync,all_squash,anonuid=1000,anongid=1000,no_subtree_check)。 /nfs 表示本机挂载路径，可替换为用户指定目录

2. 执行命令 exportfs -a -r，使/etc/exports 生效。

开发板端：

1. 创建需要挂载的目录：mkdir -p /mnt。

2. mount -t nfs {PC端IP}:/nfs /mnt -o nolock。

完成将PC端的/nfs文件夹挂载至板端/mnt文件夹。按照此方式，将包含预处理数据的文件夹挂载至板端，并将/data目录软链接至板端/ptq目录下，与/script同级目录。

模型推理

挂载完数据后，请登录开发板，开发板登录方法，请阅读 开发板登录 章节内容，登录成功后，执行 fcos_efficientnetb0/ 目录下的accuracy.sh脚本，如下图所示：

root@x3sdbx3-samsung2G-3200:/userdata/ptq/script/detection/fcos# sh accuracy.sh
../../aarch64/bin/example --config_file=workflow_accuracy.json --log_level=2
...
I0118 14:02:43.635543 24783 ptq_fcos_post_process_method.cc:157] PTQFcosPostProcessMethod DoProcess finished,
predict result: [{"bbox":[-1.518860,71.691170,574.934631,638.294922],"prob":0.750647,"label":21,"class_name":"
I0118 14:02:43.635716 24782 ptq_fcos_post_process_method.cc:150] PostProcess success!
I0118 14:02:43.636156 24782 ptq_fcos_post_process_method.cc:152] release output tensor success!
I0118 14:02:43.636204 24782 ptq_fcos_post_process_method.cc:157] PTQFcosPostProcessMethod DoProcess finished,
predict result: [{"bbox":[3.432283,164.936249,157.480042,264.276825],"prob":0.544454,"label":62,"class_name":"
...

开发板端程序会在当前目录生成 eval.log 文件，该文件就是预测结果文件。

精度计算

注解

精度计算部分请在开发机模型转换的环境下操作

精度计算的脚本在 python_tools 目录下，其中 accuracy_tools 中的： cls_eval.py是用来计算分类模型的精度； coco_det_eval.py是用来计算使用COCO数据集评测的检测模型的精度； parsing_eval.py是用来计算使用Cityscapes数据集评测的分割模型的精度。 voc_det_eval.py是用来计算使用VOC数据集评测的检测模型的精度。

• 分类模型

使用CIFAR-10数据集和ImageNet数据集的分类模型计算方式如下：

#!/bin/sh

python3 cls_eval.py --log_file=eval.log --gt_file=val.txt

注解

• log_file：分类模型的预测结果文件。

• gt_file：CIFAR-10和ImageNet数据集的标注文件。

• 检测模型

使用COCO数据集的检测模型精度计算方式如下：

#!/bin/sh

python3 coco_det_eval.py --eval_result_path=eval.log --annotation_path=instances_val2017.json

注解

• eval_result_path：检测模型的预测结果文件。

• annotation_path：COCO数据集的标注文件。

使用VOC数据集的检测模型精度计算方式如下：

#!/bin/sh

python3 voc_det_eval.py --eval_result_path=eval.log --annotation_path=../Annotations --val_txt_path=../val.txt

注解

• eval_result_path：检测模型的预测结果文件。

• annotation_path：VOC数据集的标注文件。

• val_txt_path：VOC数据集中 ../ImageSets/Main 文件夹下的val.txt文件。

• 分割模型

使用Cityscapes数据集的分割模型精度计算方式如下：

#!/bin/sh

python3 parsing_eval.py --log_file=eval.log --gt_path=cityscapes/gtFine/val

注解

• log_file：分割模型的预测结果文件。

• gt_path：Cityscapes数据集的标注文件。

模型集成

模型后处理集成主要有2个步骤，以centernet_resnet50模型集成为例：

1. 增加后处理文件 ptq_centernet_post_process_method.cc，以及头文件 ptq_centernet_post_process_method.h。

2. 增加模型运行脚本及配置文件。

后处理文件添加

后处理代码文件可直接复用src/method目录下任意后处理文件，主要修改 InitFromJsonString 函数，以及 PostProcess 函数即可。

InitFromJsonString 函数主要是读取workflow.json中的后处理相关的参数配置，用户可自定义设置相应的输入参数。 PostProcess 函数主要完成后处理的逻辑。

后处理.cc文件放置于 ai_benchmark/code/src/method/ 路径下， .h头文件放置于 ai_benchmark/code/include/method/ 路径下：

+---ai_benchmark
| +---code                                  # 示例源码
| | +---include
| | | +---method                            # 在此文件夹中添加头文件
| | | | +---ptq_centernet_post_process_method.h
| | | | +---.....
| | | | +---ptq_yolo5_post_process_method.h
| | +---src
| | | +---method                            # 在此文件夹中添加后处理.cc文件
| | | | | +---ptq_centernet_post_process_method.cc
| | | | | +---....
| | | | | +---ptq_yolo5_post_process_method.cc

增加模型运行脚本及配置文件

脚本目录结构如下：

+---ai_benchmark
| +---xj3/ptq/script                                    # 示例脚本文件夹
| | +---detection
| | | +---centernet_resnet50
| | | | +---accuracy.sh                                # 精度测试脚本
| | | | +---fps.sh                                     # 性能测试脚本
| | | | +---latency.sh                                 # 单帧延时示例脚本
| | | | +---workflow_accuracy.json                     # 精度配置文件
| | | | +---workflow_fps.json                          # 性能配置文件
| | | | +---workflow_latency.json                      # 单帧延时配置文件

辅助工具和常用操作

日志系统使用说明

​日志系统主要包括 示例日志 和 模型推理API DNN日志 两部分。 其中示例日志是指交付包示例代码中的应用日志；DNN日志是指lib dnn库中的日志。 用户根据不同的需求可以获取不同的日志。

示例日志

1. 日志等级。示例日志主要采用glog中的vlog，主要分为四个自定义等级：

• 0 (SYSTEM)，该等级主要用来输出报错信息；

• 1 (REPORT)，该等级在示例代码中主要用来输出性能数据；

• 2 (DETAIL)，该等级在示例代码中主要用来输出系统当前状态信息；

• 3 (DEBUG)，该等级在示例代码中主要用来输出调试信息。 日志等级设置规则：假设设置了级别为 P，如果发生了一个级别 Q 比 P 低， 则可以启动，否则屏蔽掉；默认DEBUG>DETAIL>REPORT>SYSTEM。

2. 日志等级设置。通过 log_level 参数来设置日志等级，在运行示例的时候，指定 log_level 参数来设置等级， 比如指定 log_level=0，即输出SYSTEM日志；如果指定 log_level=3， 则输出DEBUG、DETAIL、REPORT和SYSTEM日志。

模型推理API DNN日志

关于模型推理 DNN API日志的配置，请阅读 《模型推理接口使用说明》 文档中的 配置信息 一节内容

算子耗时说明

​对模型算子(OP)性能的统计是通过设置 HB_DNN_PROFILER_LOG_PATH 环境变量实现的，本节介绍模型的推理性能分析，有助于开发者了解模型的真实推理性能情况。 对该变量的类型和取值说明如下：

注解

export HB_DNN_PROFILER_LOG_PATH=${path}：表示OP节点dump的输出路径，程序正常运行完退出后，产生profiler.log文件。

• 示例说明

以下代码块以mobilenetv1模型为例，开启单个线程同时RunModel，设置 export HB_DNN_PROFILER_LOG_PATH=./，则统计输出的信息如下：

{
  "perf_result": {
    "FPS": 677.6192525182025,
    "average_latency": 11.506142616271973
  },
  "running_condition": {
    "core_id": 0,
    "frame_count": 200,
    "model_name": "mobilenetv1_224x224_nv12",
    "run_time": 295.151,
    "thread_num": 1
  }
}
***
{
  "chip_latency": {
    "BPU_inference_time_cost": {
      "avg_time": 11.09122,
      "max_time": 11.54,
      "min_time": 3.439
    },
    "CPU_inference_time_cost": {
      "avg_time": 0.18836999999999998,
      "max_time": 0.4630000000000001,
      "min_time": 0.127
    }
  },
  "model_latency": {
    "BPU_MOBILENET_subgraph_0": {
      "avg_time": 11.09122,
      "max_time": 11.54,
      "min_time": 3.439
    },
    "Dequantize_fc7_1_HzDequantize": {
      "avg_time": 0.07884999999999999,
      "max_time": 0.158,
      "min_time": 0.068
    },
    "MOBILENET_subgraph_0_output_layout_convert": {
      "avg_time": 0.018765,
      "max_time": 0.08,
      "min_time": 0.01
    },
    "Preprocess": {
      "avg_time": 0.0065,
      "max_time": 0.058,
      "min_time": 0.003
    },
    "Softmax_prob": {
      "avg_time": 0.084255,
      "max_time": 0.167,
      "min_time": 0.046
    }
  },
  "task_latency": {
    "TaskPendingTime": {
      "avg_time": 0.029375,
      "max_time": 0.059,
      "min_time": 0.009
    },
    "TaskRunningTime": {
      "avg_time": 11.40324,
      "max_time": 11.801,
      "min_time": 4.008
    }
  }
}

以上输出了 model_latency 和 task_latency。其中model_latency中输出了模型每个OP运行所需要的耗时情况，task_latency中输出了模型运行中各个task模块的耗时情况。

注解

程序只有正常退出才会输出profiler.log文件。

dump工具

本节主要介绍dump工具的开启方法，一般不需要关注，只有在模型精度异常情况时开启使用。 ​通过设置 export HB_DNN_DUMP_PATH=${path} 这个环境变量，可以dump出模型推理过程中每个节点的输入和输出, 根据dump的输出结果，可以排查模型推理在开发机模拟器和开发板是否存在一致性问题：即相同模型，相同输入，开发板和开发机模拟器的输出结果是否完全相同。