作者:殷忆枫

RV1126笔记二十一:车辆颜色识别

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

一、介绍

在学习RV1126的过程中,测试了yolov5可以实现物体检测,物体目标识别等功能,Rock-X也自带了车牌识别功能,具体可以了解下正点原子的资料,里面有详细的介绍,这里介绍一个如何识别车辆颜色。只是提供一个思路,效果不是很好,还需要自己优化。

有多种方法可以识别:

一、使用yolov5训练一个,这个如果有大量的图集,精度应该还可以。

二、使用opencv方式,利用rk自带的rknn模型,识别出车辆后,通过opencv算法,提取车辆颜色。

这篇主要介绍的是opencv的算法,算法可以自己网上找,或自己写,网上的精度都普遍不高,还不如自己写一个。算法是自己写的,所以这里不公开。只提供一个测试版本。主要还是一个思路,像网上的安全帽识别,可以用yolov5,也可以用opencv训练一个模型,过程只是一个过程,结果为王。

三、流程图

三、代码解析

 1、初始化

lock