arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2103.07461cs.CV

Probabilistic two-stage detection

UT Austin(德克萨斯大学奥斯汀分校);Intel Labs(英特尔实验室)

Xingyi Zhou, Vladlen Koltun, Philipp Krähenbühl

英文摘要:

We develop a probabilistic interpretation of two-stage object detection. We show that this probabilistic interpretation motivates a number of common empirical training practices. It also suggests changes to two-stage detection pipelines. Specifically, the first stage should infer proper object-vs-background likelihoods, which should then inform the overall score of the detector. A standard region proposal network (RPN) cannot infer this likelihood sufficiently well, but many one-stage detectors can. We show how to build a probabilistic two-stage detector from any state-of-the-art one-stage detector. The resulting detectors are faster and more accurate than both their one- and two-stage precursors. Our detector achieves 56.4 mAP on COCO test-dev with single-scale testing, outperforming all published results. Using a lightweight backbone, our detector achieves 49.2 mAP on COCO at 33 fps on a Titan Xp, outperforming the popular YOLOv4 model.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑