项目要求:
1.将驾驶场景分为三类:白天,夜晚-有光照,夜晚-无光照。
2.分类中有争议的场景定义:光线较暗场景
有路灯杆:1.路灯点亮--夜晚有照明。 2.路灯未亮--白天。
无路灯杆:3.夜晚无照明
数据集来源:nuScenes数据集
数据集已绑定资源,可下载
目录
1.数据集
分为三类,分别用三个英文代替:
白天-day,夜晚有光照-night,夜晚无光照-dark
可以采用爬虫或nuScenes数据集来制作所需数据集。
使用split.py将制作好的数据集划分为训练集与验证集
2.训练
夜晚无光照的图片比较少,只有200张,其他类别各有1000张左右
修改路径运行classify里的train.py开始训练
训练了50次,结果保存在runs中
3.测试
合成了一个测试视频,几乎包含全部情况
AllLight