监督学习—从好的label开始

本文通过对比实验展示了标注质量对2D目标检测模型性能的显著影响。实验使用了不同质量的KITTI数据集标注,训练faster-rcnn模型,并在Waymo和Cityscape数据集上进行测试。结果显示,精准的标注能训练出更好的模型,提高IoU阈值时,高质标注模型的性能下降幅度较小。此外,文章还讨论了数据集格式统一、样本筛选和类别统一等实验中遇到的问题。
摘要由CSDN通过智能技术生成

在大家的算法工作中,会训练各种各样的模型。导致大家模型实际应用效果不够好的原因有很多,一些常见的原因主要是模型结构不合理 、损失函数不合理 、超参数设置不合理,但除了这些原因,我想最核心的一个原因是数据的质量本身。

相信每一个自动驾驶行业的开发者对此都是会深有体会的,Lyft 团队在CVPR的presentation上就发出了“High quality labeled data is the key”的感慨。这也是本篇文章想要着重强调的主题。

 

下面通过一个实验来让大家直观感受一下“标注质量对模型训练性能的影响”

实验主题:不同质量标注对模型性能影响的对比实验

实验框架:

图中左侧是训练的过程,右侧是测试的过程

训练的部分逻辑:

首先是将KITTI数据集的图片搭配原生Original label可以得到一个Original KITTI dataset,相应地再将图片搭配Graviti label可以得到一个Graviti KITTI dataset;然后分别使用这两个数据集各自训练一个2D目标检测的模型,这里使用的是经典的faster-rcnn模型,得到的模型分别叫做Original model和Graviti model。

测试部分的逻辑:

使用了两个标注比较精准的第三方数据集;Waymo和Cityscape---来对两个模型分别进行测试。

数据处理:

先介绍一下本实验使用到的训练集和测试集:

(1)Waymo 数据集是由Waymo公司的自动驾驶汽车采集的多传感器自动驾驶数据集 它的相机数据是连续采集的,其中一百个场景提供了2D框标注。

(2&#x

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值