修改后处理的头文件

头文件路径：

rknn/rknpu-master/rknn/rknn_api/examples/rknn_yolov5_demo/include/postprocess.h

该头文件中一部分代码如下：

#ifndef _RKNN_ZERO_COPY_DEMO_POSTPROCESS_H_
#define _RKNN_ZERO_COPY_DEMO_POSTPROCESS_H_

#include <stdint.h>

#define OBJ_NAME_MAX_SIZE 16
#define OBJ_NUMB_MAX_SIZE 64
#define OBJ_CLASS_NUM     5
#define NMS_THRESH        0.6
#define BOX_THRESH        0.5
#define PROP_BOX_SIZE     (5+OBJ_CLASS_NUM)

修改【#define OBJ_CLASS_NUM 5】，将类别数修改为你的模型能识别的实际类别数量。

修改类别名称指示文件

类别指示文件路径：

rknn/rknpu-master/rknn/rknn_api/examples/rknn_yolov5_demo/model/coco_80_labels_list.txt

然后打开文件，按照你标注数据集时，标签的排列顺序，把各标签名称写入文件，如下例:

paper
plastic
metal
clothing
other

运行生成脚本

脚本所在目录

/17106/czs/rknn/rknpu-master/rknn/rknn_api/examples/rknn_yolov5_demo

执行命令:

bash build.sh