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

SOLOv2: Dynamic and Fast Instance Segmentation

The University of Adelaide(阿德莱德大学);Tongji University(同济大学);ByteDance AI Lab(字节跳动人工智能实验室)

Xinlong Wang, Rufeng Zhang, Tao Kong, Lei Li, Chunhua Shen

英文摘要:

In this work, we aim at building a simple, direct, and fast instance segmentation framework with strong performance. We follow the principle of the SOLO method of Wang et al. "SOLO: segmenting objects by locations". Importantly, we take one step further by dynamically learning the mask head of the object segmenter such that the mask head is conditioned on the location. Specifically, the mask branch is decoupled into a mask kernel branch and mask feature branch, which are responsible for learning the convolution kernel and the convolved features respectively. Moreover, we propose Matrix NMS (non maximum suppression) to significantly reduce the inference time overhead due to NMS of masks. Our Matrix NMS performs NMS with parallel matrix operations in one shot, and yields better results. We demonstrate a simple direct instance segmentation system, outperforming a few state-of-the-art methods in both speed and accuracy. A light-weight version of SOLOv2 executes at 31.3 FPS and yields 37.1% AP. Moreover, our state-of-the-art results in object detection (from our mask byproduct) and panoptic segmentation show the potential to serve as a new strong baseline for many instance-level recognition tasks besides instance segmentation. Code is available at: https://git.io/AdelaiDet

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

↑