AI搬运 · 9月16日

YOLO窥见黑夜|YOLO in the Dark让黑夜里的目标检测成为可能

来源:AI人工智能初学者
作者:ChaucerG

640.png

对于处理新的视觉任务,则需要额外的数据集,这需要花费大量精力。本文提出了一种域自适应的新方法,该方法可以比创建附加数据集更轻松地合并多个模型。该方法使用glue层和生成模型合并了不同领域中的预训练模型,该模型提供了潜在特征来训练glue层而无需其他数据集。我们还提出了从预先训练的模型中通过知识蒸馏创建的生成模型。它还允许重用数据集来创建潜在特征以训练胶合层。我们将此方法应用于弱光条件下的目标检测。“Dark YOLO”包含两个模型,“学习如何在黑暗中看”和YOLO。与其它方法相比,“Dark YOLO”花费更少的计算资源。

1、简介

在光线弱的情况下进行视觉任务是一个比较困难的课题。Short-Exposure图像没有足够的特征进行视觉处理,而图像的亮度增强会引起噪声进而影响视觉任务。相比之下,Long-Exposure图像也含有噪声,由于运动模糊而影响视觉任务。
前人一些工作可以总结为以下三点:

  • 1、通过制作附加数据集(比如说the See-in-the-Dark dataset)的方式来缓解这个问题,让数据集中尽可能的包含多种Exposure情况下的图像,但是带来的弊端就是,需要多余的人力来解决,同时不是一个end to end的模型。
  • 2、通过知识蒸馏的方式解决这个问题,相对于构造附加数据集的方式而言是个更好的处理方式;
  • 3、使用无监督学习的方式来学习Domain Gap,进而学到Domain Adaption特性,进而提升在不同Domain之间的适应性,方便Domain的迁移。

2、本文方法

本文主要还是基于前面提到的知识蒸馏的方法进行的设计,提出了YOLO in the Dark模型。640-1.png

图2 使用领域适应方法融合了在Domain A和Domain B训练的2个模型

这里model A从一个RAW图片中预测一个RGB图片,然后model B从RGB图像中预测目标的位置和类别,完成model A和B的训练后,以潜在特征A和B的边界提取模型fragments。new model由model A和B的以潜在特征A和B的边界提取模型fragments通过一层粘合层(Glue Layer)组合而成。

Glue Layer层可以将模型fragments中的latent feature A转换为latent feature B。SID模型在低光图像上有比较好的效果,因此对model A使用SID模型。还使用目标检测模型YOLO对于model B。

2.1、Domain Adaption的生成模型

通过图2也可以看出来训练Gule Layer需要Domain A+B的数据,然而制作一个这样的数据集需要很大的功夫,所以作者在这里选择了知识蒸馏的方法来定义一个生成模型进而来训练Gule Layer

640-2.png

图3 知识蒸馏方案.红色模块G2e是Student Model

图3 解释了知识蒸馏的方案,其中的生成模型输出的latent feature A来自于,SID模型是编解码的结构,因此生成模型可以作为解码器的映射函数;同时作为Teacher Model同时使用来自SID数据集和模型的数据样本对来训练Student Model

image.png

640-3.png

图4显示的Gule Laye的Latent Feature SID编码器

image.png

640-4.png

图5显示了使用SID编码器的潜在特性重新构建的RGB图像。(a)展示了使用所有特性重构的图像。这些图像的峰值信噪比(PSNR)为31.81,具有结构特征相对于原始图像的相似性(SSIM)为0.752。(b)、(c)和(d)为使用较少特征重建的图像,去除了高空间频率信息。这些图象的质量比图象(a)的质量差。但是为了检测物体,必须识别出物体的具体形状,因此文章决定使用所有的Latent Feature用于Gule Layer。

640-5.png

image.png

2.2、Training environment

图7(a)显示了环境的完整视图,其中点边界显示了用于训练新模型的部分,其中Gule Layer是模型训练的目标;而训练该模块使用的RGB图像数据即是G2e编码器通过知识蒸馏的方法得到的。整体的训练环境还是基于原生的YOLO模型,使用和G2e编码器一样的RGB数据,这里使用的是COCO数据集进行训练。

640.png

image.png

3、实验结果

图8显示了SID数据集的对象检测结果。图8(a)是原始YOLO模型使用亮度增强的RGB图像得到的检测结果。RGB图像的亮度增强使得原始YOLO模型更容易检测到目标。因此原始的YOLO模型可以很好地检测图像中的对象。但是,该模型无法检测到图像中的目标。这是因为亮度增强增加了噪音,影响了模型的推断。而本文提出的方法可以直接检测RAW图像中的目标。检测结果如图b1和b2所示。图像c1和c2是标签,标签是通过原始YOLO模型使用SID ground truth(长曝光)图像检测得到。在图像b1中,提出的模型表现得和原始的YOLO模型(图像a1)一样好。此外,所提出的模型可以检测图像b2中的目标。

640-1.png
640-2.png

References

[1] YOLO in the Dark - Domain Adaptation Method for Merging Multiple Models

声明:转载请说明出处

推荐专栏文章


更多嵌入式AI算法部署等请关注极术嵌入式AI专栏
1 阅读 238
推荐阅读
0 条评论
关注数
11049
内容数
157
嵌入式端AI,包括AI算法在推理框架Tengine,MNN,NCNN,PaddlePaddle及相关芯片上的实现。欢迎加入微信交流群,微信号:gg15319381845(备注:嵌入式)
目录
极术微信服务号
关注极术微信号
实时接收点赞提醒和评论通知
Arm中国学堂公众号
关注Arm中国学堂
实时获取免费 Arm 教学资源信息
Arm中国招聘公众号
关注Arm中国招聘
实时获取 Arm 中国职位信息