RV1126笔记二十:吸烟行为检测及部署<七>

若该文为原创文章,转载请注明原文出处。

部署到RV1126,Demo测试

一、介绍

通过训练转换后,得到了RKNN模型,接下来使用rknn_model_zoo里自带的C demo来测试模型是不是可以在RV1126上运行。

C demo直接编译是编译不过的,需要自己移植.

根据C demo提供的README,可以看出,采用的是单图和多图方式,图片格式是JPG,这不符合我的测试,想采用的是bmp,所以需要修改一些代码。 

RV1126笔记二十:吸烟行为检测及部署<七>_第1张图片

二、流程图

RV1126笔记二十:吸烟行为检测及部署<七>_第2张图片

三、代码移植过程

代码移植我是以EASY EAI里的yolov5 C Demo来移植的,原因是这个C Demo不使用opencv,加载的BMP图片。

移植相对简单,主要修改3个函数:main.cc yolo.cc yolo.h,结构目录如下,我使用的是Makefile方式。 

RV1126笔记二十:吸烟行为检测及部署<七>_第3张图片

説明:

3rdparty:第三方文件,直接拷贝yolov5 C

你可能感兴趣的:(RV1126项目实战,计算机视觉,opencv,人工智能)