基于交通灯数据集的端到端分类