4 个月前

面向领域适应的Faster R-CNN用于野外目标检测

面向领域适应的Faster R-CNN用于野外目标检测

摘要

目标检测通常假设训练数据和测试数据来自相同的分布,然而在实际应用中这一假设并不总是成立。这种分布差异会导致性能显著下降。在本研究中,我们旨在提高目标检测在不同域间的鲁棒性。我们从两个层面应对域偏移问题:1)图像层面的偏移,如图像风格、光照等;2)实例层面的偏移,如目标外观、大小等。我们的方法基于最新的先进Faster R-CNN模型,并设计了两个域适应组件,分别用于图像层面和实例层面,以减少域间差异。这两个域适应组件基于H-散度理论,并通过对抗训练方式学习一个域分类器来实现。不同层次的域分类器进一步通过一致性正则化强化,以学习Faster R-CNN模型中的域不变区域提议网络(Region Proposal Network, RPN)。我们使用多个数据集(包括Cityscapes、KITTI、SIM10K等)对所提出的新方法进行了评估。结果表明,该方法在各种域偏移场景下均能有效提高目标检测的鲁棒性。

代码仓库

shreyasrajesh/DA-Object-Detection
pytorch
GitHub 中提及
harsh-99/SCL
pytorch
GitHub 中提及
yuhuayc/da-faster-rcnn
官方
pytorch
GitHub 中提及
jinlong17/da-detect
pytorch
GitHub 中提及
krumo/Detectron-DA-Faster-RCNN
pytorch
GitHub 中提及

基准测试

基准方法指标
image-to-image-translation-on-cityscapes-toFRCNN in the wild
mAP: 27.6
unsupervised-domain-adaptation-on-cityscapes-1DA-Faster
mAP@0.5: 26.1

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
面向领域适应的Faster R-CNN用于野外目标检测 | 论文 | HyperAI超神经