/yolov5-TT100k

基于yolov5的交通标志牌识别项目,使用tt100k数据集

Primary LanguagePythonGNU Affero General Public License v3.0AGPL-3.0

说明

本项目基于YOLOv5的框架,主要的功能是实现基于TT100K数据集的交通信号灯与标志牌识别任务。

在YOLOv5的框架下,我所作的主要贡献之一是提供了将TT100K原本的数据集转化为YOLOv5训练所需数据集的格式,具体代码见demox.py,其中x表示数字,之所以会有不同版本的demo是基于不同的情况;共享之二是改进了原本YOLOv5的网络结构,使得改进后的网络对于小目标物体的检测能力得到加强,同时使得网络轻量化,大幅降低了网络参数。

下面是一些我们的实验结果: image-20231229170422142

image-20231229170436294

image-20231229170444261