干货 | YOLOV5 训练自动驾驶数据集,并转Tensorrt,收藏!
AI视线 点击关注上方“AI深度学习视线”,并“星标”公号 技术硬文,第一时间送达! 准备数据集 环境配置 配置文件修改 训练 推理 转Tensorrt 1 准备数据集 1.1 BDD数据集 BDD100K是最大的开放式驾驶视频数据集之一,其中包含10万个视频和10个任务,目的是方便评估自动驾驶图像识别算法的的进展。每个高分辨率视频一共40秒。该数据集包括超过1000个小时的驾驶数据,总共超过1亿帧。这些视频带有GPU / IMU数据以获取轨迹信息。该数据集具有地理,环境和天气多样性,从而能让模型能够识别多种场景,具备更多的泛化能力。这些丰富的户外场景和复杂的车辆运动使感知任务更具挑战性。该数据集上的任务包括图像标记,车道检测,可驾驶区域分割,道路对象检测,语义分割,实例分割,多对象检测跟踪,多对象分割跟踪,领域自适应和模仿学习。我们可以在BDD100K数据网站上下载数据。 Bdd100k的标签是由Scalabel生成的JSON格式。- labels [ ]: - id: int32 - category: string (classification) - manualShape: ...


