利用GPU在高分辨率4K与8K视频中实现快速准确的目标检测
计算机视觉与模式识别
2019-02-01 v1 机器学习
摘要
机器学习在计算机视觉任务（如目标检测）上取得了诸多成就，但传统使用的模型处理的是相对低分辨率图像。录制设备的分辨率逐渐提高，对处理高分辨率数据的新方法需求上升。我们提出一种注意力流水线方法，对每幅图像或视频帧在粗略与精细分辨率下使用两阶段评估，以限制必要的评估总数。两个阶段我们都使用了快速目标检测模型YOLO v2。我们已用代码实现模型，将工作分配至GPU。我们在4K视频上达到3-6 fps、8K视频上2 fps的平均性能，同时保持高准确率。
引用
@article{arxiv.1810.10551,
title = {Fast and accurate object detection in high resolution 4K and 8K video using GPUs},
author = {Vít Růžička and Franz Franchetti},
journal= {arXiv preprint arXiv:1810.10551},
year = {2019}
}
备注
6 pages, 12 figures, Best Paper Finalist at IEEE High Performance Extreme Computing Conference (HPEC) 2018; copyright 2018 IEEE; (DOI will be filled when known)