中文

YOLOv4:目标检测的最优速度与精度

计算机视觉与模式识别 2020-04-24 v1 图像与视频处理

摘要

有大量被认为能提高卷积神经网络(CNN)精度的特征。需要在大型数据集上对这些特征的组合进行实际测试,并对结果进行理论论证。某些特征仅适用于特定模型和特定问题,或仅适用于小规模数据集;而某些特征,如批归一化和残差连接,适用于绝大多数模型、任务和数据集。我们假设此类通用特征包括加权残差连接、跨阶段部分连接、跨小批量归一化、自对抗训练和Mish激活函数。我们使用了新特征:WRC、CSP、CmBN、SAT、Mish激活、Mosaic数据增强、CmBN、DropBlock正则化和CIoU损失,并将其中部分组合以实现最先进的结果:在Tesla V100上以约65 FPS的实时速度,MS COCO数据集上达到43.5% AP(65.7% AP50)。源代码位于 https://github.com/AlexeyAB/darknet

关键词

引用

@article{arxiv.2004.10934,
  title  = {YOLOv4: Optimal Speed and Accuracy of Object Detection},
  author = {Alexey Bochkovskiy and Chien-Yao Wang and Hong-Yuan Mark Liao},
  journal= {arXiv preprint arXiv:2004.10934},
  year   = {2020}
}