Windows10下使用keras-yolov3+CCTSDB(中国交通数据集)进行道路元素检测

Windows10下使用keras-yolov3+CCTSDB(中国交通数据集)进行道路元素检测

2019年10月份的时候做了相关实验，最近闲来无事便写成博客

CCTSDB数据集来源

CSUST Chinese Traffic Sign Detection Benchmark 中国交通数据集由长沙理工大学综合交通运输大数据智能处理湖南省重点实验室张建明老师团队制作完成。来源于 A Real-Time Chinese Traffic Sign Detection Algorithm Based on Modified YOLOv2一文提出的训练数据集。
下载地址：参考博客https://blog.csdn.net/dong_ma/article/details/84339007
数据集本来已经标注，但是png格式，而我要在Windows10下面进行训练，所以我用格式工厂把我需要的图片转换成jpg格式。
这里我第一次用了80张图片，第二次，结果失败了，测试的时候没有检测框，然后我在数据集里精选了500张图片，从00001到00500重新命名，如下图所示

LabelImg进行标注
然后使用LabelImg进行标注，生成.xml文件，共标注6类。LabelImg的安装参考博客：https://blog.csdn.net/enjoy_endless/article/details/80803149

                            限速：speed limit
                            警告：warning
                            提示：mandatory
                            禁止：prohibitory
                            红灯：red light
                            绿灯：green light

voc2007数据集制作
下载yolov3代码：https://github.com/qqwweee/keras-yolo3， 具体制作过程可以看下面博客，写的特别好
https://blog.csdn.net/u012746060/article/details/81183006制作完以后，修改相应的参数就可以进行训练了
训练测试
用了500张图片，epochs=500,训练集289张，验证集193张，检测效果也不是好。

然后我修改了一下，样本500张，epochs=1000，训练集：450张，验证集50张，检测效果如下

想试一下训练的模型咋样，就拿了几张自己拍的图片测试一下

  测试结果如下：

可以看出图中右下角一个红灯没有检测出来，还有指示标志也被错误识别为禁止标志

这里有个红灯没有检测出来，提示标志：mandatory虽然检测出来了，但iou只有0.47，我设置的是0.5，所以还是检测失败了。

代码链接：https://download.csdn.net/download/weixin_42097734/12434663

第一次写博客，写的不好，请大家见谅。