使用硬分类算法对红绿灯进行分类
2022-08-24 19:05:56 9KB 红绿灯分类
1
整个项目源码: 引言 前面我们讲完交通标志的识别,现在我们开始尝试来实现交通信号灯的识别 接下来我们将按照自己的思路来实现并完善整个Project. 在这个项目中,我们使用HSV色彩空间来识别交通灯,可以改善及提高的地方: 可以采用Faster-RCNN或SSD来实现交通灯的识别 首先我们第一步是导入数据,并在RGB及HSV色彩空间可视化部分数据。这里的数据,我们采用的图片, 总共三类:红绿黄,1187张图片,其中,723张红色交通灯图片,429张绿色交通灯图片,35张黄色交通灯图片。 导入库 # import some libs import cv2 import os import glob import random import numpy as np import matplotlib.pyplot as plt import matplotlib.image as mpimg
2022-05-11 23:38:53 4.16MB JupyterNotebook
1