实现夜间多目标追踪,并开放低光多目标追踪数据集LMOT
LTrack:实现夜间多目标追踪,并开放低光多目标追踪数据集LMOT
摘要
低光场景在现实应用中很常见(例如,夜间的自动驾驶和监控)。最近,多目标跟踪在各种实际用例中受到了很多关注,但黑暗场景中的多目标跟踪却很少被考虑。在本文中,我们专注于黑暗场景中的多目标跟踪。为了应对数据集的缺乏,我们首先建立了一个低光多目标跟踪(LMOT)数据集。LMOT提供了通过我们的双摄像头系统捕获的对齐良好的低光视频对,并为所有视频提供高质量的多目标跟踪注释。然后,我们提出了一种低光多目标跟踪方法,称为LTrack。我们引入了自适应低通下采样模块,以增强图像中传感器噪声之外的低频成分。降解抑制学习策略使模型能够在噪声干扰和图像质量降级下学习不变的信息。这些组件提高了黑暗场景中多目标跟踪的鲁棒性。我们对LMOT数据集和提出的LTrack方法进行了综合分析。实验结果证明了所提方法的优越性及其在夜间低光场景中的竞争力。
代码地址:https://github.com/ying-fu/LMOT
附赠自动驾驶最全的学习资料和量产经验:链接
Introduction
多目标跟踪(MOT)旨在定位和关联视频序列中的多个对象。它被广泛应用于许多下游应用中,例如视频识别、自动驾驶和监控。最近,针对各种实际用例的多目标跟踪受到了很多关注,极大地促进了MOT的发展。然而,这些工作主要针对高质量输入,忽略了现实世界中普遍存在的低光场景。出于这个动机,我们研究了黑暗场景中的多目标跟踪。
在本文中,我们构建了一个专门设计来解决黑暗场景中多目标跟踪挑战的低光多目标跟踪数据集(LMOT)。为此,我们开发了一套能够同时捕获亮光和低光视频帧的双摄像头系统。这些视频对在空间和时间维度上高度对齐,提供了两个关键好处。首先,它使我们能够在亮光视频上进行注释,从而获得高质量的注释。其次,亮光视频可以在训练阶段提供额外的监督信息,显著增强黑暗场景中的表现。经过仔细注释,我们收集了32个视频序列(是MOT17的2.3倍),超过35K帧(是MOT17的3.1倍),和超过815K个边界框(是MOT17的2.8倍)。RAW数据是图像传感器的输出,是图像信号处理器(ISP)的输入数据。它保存了所有从图像传感器获取的信息,这对于在黑暗场景中捕捉对象信息至关重要。因此,我们收集了LMOT的RAW视频。
此外,我们提出了一种低光多目标跟踪方法,称为LTrack。低光视频的特点是传感器噪声显著且图像质量差,这显著降低了浅层和深层特征表示,导致跟踪性能下降。我们观察到,低光图像中的传感器噪声与对抗攻有相似之处。为了解决这个问题,我们的主要思想是在噪声干扰和质量下降下学习不变的语义信息。我们提出了自适应低通下采样模块(ALD)。它通过空间低通卷积从图像中提取低频成分,排除噪声,并自适应增强特征图。我们还提出了降解抑制学习策略(DSL),利用配对的低光视频帮助模型抑制噪声干扰,并在特征域中鼓励图像内容响应。我们对LMOT数据集进行了综合分析,并验证了LTrack在真实夜间场景中的优越性。
Method
低光多目标跟踪数据集
在本节中,我们首先介绍我们的双摄像头系统以及收集和注释低光多目标跟踪(LMOT)数据集的详细信息。然后,我们分析了LMOT数据集的统计特征。
1 数据集构建
多目标跟踪需要动态场景和对象视频。为了收集低光视频进行多目标跟踪,我们构建了一套双摄像头系统,如图1所示。它可以同时捕获配对的低光和亮光视频。其主要组件包括分光器、中性密度(ND)滤光片和两个FLIR Grasshopper3 GS3-U3-23S6C摄像头。分光器将进入的光分成两条独立的光路。这种布置使一台摄像机直接捕捉亮光图像,而另一台摄像机记录低光图像,ND滤光片衰减光强度。为了确保视频帧的时间同步,我们采用硬件接口触发摄像机曝光事件。此外,为了避免帧丢失,我们的双摄像头系统使用两个独立的硬件接口进行数据传输,并配备了高速固态硬盘。得益于精确校准,我们的双摄像头系统能够实时捕捉动态场景和对象的配对低光和亮光视频。关于我们双摄像头系统的更多详细信息,请参见补充材料。
我们将视频帧保存为RAW格式,在图像处理之前,这些图像由摄像头图像信号处理器(ISP)处理。就摄像头设置而言,我们将两台摄像机的曝光时间设置为10毫秒,帧速率固定为20帧。这一设置能够避免运动模糊。我们调整了亮光摄像头的增益水平以达到最佳图像质量。低光摄像头的增益始终设置为最大值,以模拟实际场景中的低光拍摄设置。我们还收集了一个真实的低光MOT数据集(LMOT-real)来评估在真实黑暗场景中的表现。这些视频使用相同的摄像头设置单独拍摄。
我们的LMOT数据集包含各种城市户外场景,包括道路、天桥、行人和十字路口。天桥场景从高处拍摄对象,而所有其他场景均从行人视角拍摄。为了考虑摄像机运动的影响,我们引入了任意的水平旋转和垂直随机移动摄像机。图2显示了我们的LMOT数据集中两个视频序列的采样。
我们注释了六类移动对象,包括汽车、行人、自行车、摩托车、公交车和卡车。注释标签包括边界框、目标ID和目标类别。我们将亮光视频注释结果映射到低光视频中,以确保所有低光视频都具备与亮光视频相同的MOT注释。为了获得高质量注释,我们在原始亮光视频上进行半自动注释,首先通过ByteTrack检测所有目标框。然后,我们使用SORT将检测到的框与亮光视频上的跟踪结果相关联,并通过修正小的错位或遗漏来进一步提升注释质量。最终的注释由两个团队成员审阅和核实。
2 数据集分析
为了演示LMOT数据集的独特性,我们提供了其统计特征并与流行的MOT数据集进行比较。在表1中,我们总结了七个MOT数据集,包括车辆(KITTI和 UA-DETRAC)、行人(ETH、DukeMTMC和 MOT17)和混合场景(BDD100K和 PathTrack)。请注意,原始BDD100K MOT数据集没有低光场景,而PathTrack是大规模数据集,但没有低光条件。LMOT在数据量上远超其他流行MOT数据集。它包含35,481帧(MOT17的3.1倍),815,035个边界框(MOT17的2.8倍)和32个视频序列(MOT17的2.3倍)。此外,LMOT视频序列包含多种动态场景和对象。
图3显示了不同数据集中的帧速度分布。为简洁起见,我们仅显示LMOT与MOT17和KITTI的对比。与其他MOT数据集相比,LMOT中的对象速度分布更加均匀,涵盖了宽范围的运动速度。这些对象包括行人、自行车、汽车等,行人通常比汽车更慢。因此,对象速度的分布更加均匀。
图4对比了LMOT、KITTI和MOT17数据集中不同场景下的帧平均IoU(目标框与注释框的交并比)。在车辆MOT任务中,KITTI视频帧通常具有较大的目标框,其IoU较大。行人MOT任务中,MOT17视频帧的IoU较小。LMOT视频帧具有较小的IoU。这意味着LMOT视频中的目标运动速度更快。总之,LMOT数据集具有动态场景和对象、较大的数据量和高质量的注释,极具挑战性。
低光多目标跟踪
在本节中,我们提出了一种低光多目标跟踪方法(LTrack)。我们的主要思路是在噪声干扰和质量下降的情况下学习不变的语义信息。总体框架如图4所示。
1. 公式化和动机
在低光场景中,相机在单次曝光中只能捕捉到少量光子。因此,潜在的传感器噪声被放大,导致图像质量显著下降。我们观察到,直接将低光图像输入网络而不进行任何特殊设计,会导致特征图退化,显著降低模型性能。一种直接的解决方案是应用低光增强技术,专注于学习从低光图像到清晰良好照明图像的映射函数。由于这是一个高度不适定问题,学习这样的映射函数需要大量的计算和存储开销。尽管基于DNN的方法在低光增强方面取得了优异的性能,但增强后的图像对于下游任务可能并不理想。在这项工作中,我们从低光图像中进行多目标跟踪,绕过了低光增强。利用RAW视频,网络比sRGB获得了更多的原始场景信息。为了增强多目标跟踪模型的性能和鲁棒性,我们的主要思路是在噪声干扰和质量下降的情况下学习不变的语义信息。因此,我们提出了自适应低通下采样(ALD)模块,以增强图像的低频成分并过滤掉高频噪声。我们还提出了降解抑制学习策略(DSL),利用配对的低光视频帮助模型抑制图像噪声干扰,并在特征域中鼓励图像内容响应。
2. 自适应低通下采样
下采样操作在保留最重要信息的同时减少特征尺寸。低光图像中的噪声会将高频干扰引入特征图,从而误导目标信息的保留。为了减弱高频噪声的影响并增强特征图的低频部分,我们引入了空间低通卷积(SConv)来从噪声特征图中提取低频特征。
3. 降解抑制学习
由于低光图像和良好照明图像对共享相同的内容,模型应对它们表现出相同的特征响应。然而,低光图像的浅层特征充满噪声,深层特征对目标的响应较低(如图5所示)。为了解决这个问题,我们的想法是在浅层特征中抑制图像噪声,并利用良好照明的图像帮助模型学习低光图像中的干扰不变信息。
Experiments
1. 实验设置
数据集划分在构建数据集时,我们随机将视频划分为训练集、验证集和测试集,分别由11、4和11个视频组成。我们还提供了在真实夜间场景中捕获的6个视频的LMOT-real数据集。详细的数据集划分和统计信息如表2所示。
2. 低光条件下的分析
对检测器的影响
我们首先分析光照条件对检测器的影响。我们选择YOLOX作为检测器,因为它在MOT领域被广泛使用。我们使用良好照明图像(WL)、低光图像(LL)和所有图像(AL)训练检测器。然后,在良好照明和低光图像上测试它们。结果如表3所示。从表中可以看出,使用良好照明图像训练的模型在良好照明图像上取得了最佳结果,但其在低光图像上的性能显著下降。
对关联模块的影响
分析低光条件对目标关联模块的影响。运动和外观对目标关联很重要。两者都依赖于检测框来定位目标。为了分离检测器的影响,我们使用地面实况检测框作为检测框。结果如表4所示。可以看出,使用仅外观匹配在良好照明条件下取得了最佳结果,而在低光条件下使用仅外观匹配的结果非常差。这表明LMOT中的目标在特征空间中具有明显的视觉可区分性,但在低光条件下这种可区分性显著降低。在图6中,我们可视化了LMOT数据集中良好照明和低光条件下目标的外观特征。
3 在LMOT数据集上的结果
从表5中可以看出,所提出的LTrack在所有指标上都具有最佳的HOTA并且竞争力很强。例如,所提出的方法在与基线方法几乎相同的参数和计算量下提高了1.3的HOTA。这强有力地证明了所提出方法的有效性。
4 真实世界中的结果
为了验证我们的方法在夜间真实世界低光场景中的性能,我们使用OC-SORT对LMOT-real数据集上的所有方法进行了评估。如表6所示,所提出的LTrack表现明显优于所有比较方法。无论是低光增强后的跟踪方法还是RAW检测方法都遇到了泛化性问题,甚至不如Base-all。这强烈证明了我们的LTrack在真实世界黑暗场景中的鲁棒性。
5 探索与讨论
在本节中,我们对LMOT数据集和所提出的方法进行了广泛的分析和讨论。
消融研究。我们进行了消融实验,以验证我们改进的有效性,结果如表7所示。可以看出,所有改进都有效地提高了性能。其中,降解抑制学习策略表现出最显著的效果,使HOTA大约提升了1点。当同时采用所有策略时,达到了最佳效果。这证明了我们所有贡献的有效性。
RAW vs. sRGB。我们分析了输入数据格式的影响,结果如表8所示。12位的RAW格式比sRGB取得了显著更好的结果。因为RAW格式保存了更多潜在的信息,对低光场景中的多目标跟踪(MOT)很有帮助。我们还观察到,较高的位宽有利于性能,这在其他视觉任务中也有所观察。
LMOT vs. 合成数据。我们将我们的LMOT数据集与合成低光数据进行了比较,以进一步证明LMOT的价值。我们应用基于高斯-泊松的和基于物理的低光数据合成方法,从良好光照视频合成低光视频。我们在这些类型的数据上训练了我们的LTrack,并在使用LMOT-real进行的真实低光场景中评估了它们的性能。如表9所示,在我们的LMOT数据集上训练的跟踪器在真实夜间场景中表现要好得多,这强烈证明了我们的LMOT数据集的价值。
不同类别的分析。我们还分析了不同类别的表现。从表10中可以看出,汽车和巴士由于形状规则和面积较大,表现最佳。卡车的表现最差,因为它们的实例最少,使得模型难以准确识别。行人取得了相对平均的分数。自行车和摩托车的分数接近,因为它们在外观和运动模式上相似。
总结
本文的主要贡献如下:
-
使用精心构建的双摄像头系统构建了第一个低光多目标跟踪数据集。它提供了对齐良好的RAW格式低光视频和高质量的MOT注释。
-
提出了一种低光多目标跟踪方法。它利用自适应低通下采样模块和降解抑制学习策略来学习从低光视频中提取不变特征。
-
对数据集和提出的方法进行了综合分析。实验结果证明了所提方法的优越性及其在真实夜间场景中的竞争力。
引用文章:
Multi-Object Tracking in the Dark