课题:交通标志识别——设计思路与实现步骤记录

(一)准备数据集
选取两个典型的交通标志数据集(德国交通标志数据集GTSRB
比利时交通标志数据集
选取两个数据集原因
1.便于后续对测试结果的对比,以及准确率差别的分析
2.二者图片格式相似,同样的程序二者兼容,无需重新编写代码
3.二者都分为训练集和测试集,同时都有存贮着ROI特征区域信息的.csv文件,对图片处理过程相同
4.二者标签数量以及各个标签下的图片数量不同,便于对比

(二)图片处理&数据集修改完善(imagedeal.py)
1.数据库详细介绍:两个数据库的图片存储格式都为.ppm格式,每个类别文件夹(分类标签)下面除相应的交通标志外,还存有一个.scv文件,此文件用Excel可以打开,存储了该文件夹下的所有图片信息,包括图片名、长、宽、ROI特征区域的坐标(X1,Y1,X2,Y2)以及分类标签号,后面将根据csv文件中存贮的信息对图像进行处理。在这里插入图片描述
2.载入模型之前,对图片进行处理。
用imagedeal.py程序对图片进行ROI区域提取,图片大小统一为 32*32,并将图片格式转化为JPG。

(三)图片预处理(Trafficsign_Main.py)
搭建步骤:
1.导入相关库
2.定义相关参数
3.导入训练图片
4.将图片分为测试集与训练集(注:模型训练与测试来自同一Dataset,按照比例将该Dataset分为测试集与训练集)
5.读取.csv文件
6.显示所有标签中的一些样本图片(N个标签*5张)
在这里插入图片描述

7.用直方图显示每个标签中所含样本数目
在这里插入图片描述

8.对输入的图片进行预处理
(1)BGR转换为灰度图
(2)直方图均衡化
(3)图像增广
9.显示预处理后的部分图片样本
在这里插入图片描述

(四)搭建卷积神经网络模型(Trafficsign_Main.py)

模型逐层参数如下(运行程序获得):
在这里插入图片描述

(五)模型训练(Trafficsign_Main.py)
第一轮测试:
在这里插入图片描述
在这里插入图片描述
第二轮测试:
在这里插入图片描述
第五轮测试:
在这里插入图片描述
最后一轮测试:
在这里插入图片描述

(六)绘制loss曲线与accuracy曲线(Trafficsign_Main.py)
在这里插入图片描述
在这里插入图片描述

(七)模型测试、实时测试(TrafficSign_Test)
1.导入相关库
2.设置摄像头参数、激活摄像头
3.读入之前训练保存的模型
4.定义图片处理函数
(1)BGR转换为灰度图
(2)直方图均衡化
(3)图像归一化
5.定义标签与图片描述相对应的函数
6.while True中依次执行:
(1)读取图片
(2)图片处理
(3)图片标签预测
(4)显示预测结果
(5)关闭摄像头

  • 1
    点赞
  • 17
    收藏
    觉得还不错? 一键收藏
  • 8
    评论
交通标志识别系统标准完全模板HSV定位基础版-交通警令标示图片.rar 本帖最后由 hcq@@ 于 2013-7-7 11:04 编辑 1、交通标志(警示牌类),其外框一般为红色,所以定位时可采用HSV颜色模型或者RGB颜色模型来提取红色区域,通过regiongroup标记进一步提取目标区域。 HSV颜色模型:I1=Hsv; %记录颜色亮度为1的数据,以便之后再利用颜色阀值找出红色 figure,imshow;title'); BW=roicolor; %利用颜色阀值将红色对象都都显示为白色,其余都为黑色,至此完成图象的二值化 figure,imshow;title; 复制代码具体参考附件HSV提取基础版.rar RGB颜色模型 IR=double; IG=double; IB=double; for i=1:chang for j=1:kuan if 1.6*IR-IG-IB255 CA=255; else CA=1.6*IR-IG-IB; end end end figure;imshow;title; 复制代码经过对比,对于目标边界清晰的图片,RGB颜色模型的效果很不错,结果在贴图中。 2、分割出来的交通标志,为了便于二值化处理,强烈建议将背景黑色转成白色,这里标准模板只处理了20几张(考虑到可能用不到这么多的模板),如果有需要,可采取灰度—二值化—反色—擦除边缘—归一化步骤,即可处理成新的模板。代码在 附件里面: 3、识别的时候,可先建立数组,和普通的识别方法一样的,识别结果可采取msgbox的显示方法(即根据返回的匹配索引,得出结果)。if x==1 msgbox;end if x==2 msgbox;end if x==3 msgbox;end if x==4 msgbox;end 。。。。。。。同理 复制代码

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 8
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值