瑞芯微1808开发板目标识别模型运行环境的生成

1. 修改后处理的头文件

   头文件路径：

   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】，将类别数修改为你的模型能识别的实际类别数量。

2. 修改类别名称指示文件

   类别指示文件路径：

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

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

   paper
   plastic
   metal
   clothing
   other
   

3. 运行生成脚本

   脚本所在目录

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

   执行命令:

   bash build.sh