YOLACT plus plus Better Real-Time Instance Segmentation

被引:291
|
作者
Bolya, Daniel [1 ]
Zhou, Chong [2 ]
Xiao, Fanyi [2 ]
Lee, Yong Jae [2 ]
机构
[1] Georgia Inst Technol, Coll Comp, Atlanta, GA 30332 USA
[2] Univ Calif Davis, Dept Comp Sci, Davis, CA 95616 USA
关键词
Instance segmentation; real time;
D O I
10.1109/TPAMI.2020.3014297
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
We present a simple, fully-convolutional model for real-time ( > 30 fps) instance segmentation that achieves competitive results on MS COCO evaluated on a single Titan Xp, which is significantly faster than any previous state-of-the-art approach. Moreover, we obtain this result after training on only one GPU. We accomplish this by breaking instance segmentation into two parallel subtasks: (1) generating a set of prototype masks and (2) predicting per-instance mask coefficients. Then we produce instance masks by linearly combining the prototypes with the mask coefficients. We find that because this process doesn't depend on repooling, this approach produces very high-quality masks and exhibits temporal stability for free. Furthermore, we analyze the emergent behavior of our prototypes and show they learn to localize instances on their own in a translation variant manner, despite being fully-convolutional. We also propose Fast NMS, a drop-in 12 ms faster replacement for standard NMS that only has a marginal performance penalty. Finally, by incorporating deformable convolutions into the backbone network, optimizing the prediction head with better anchor scales and aspect ratios, and adding a novel fast mask re-scoring branch, our YOLACT++ model can achieve 34.1 mAP on MS COCO at 33.5 fps, which is fairly close to the state-of-the-art approaches while still running at real-time.
引用
收藏
页码:1108 / 1121
页数:14
相关论文
共 50 条
  • [21] Multi-scale YOLACT for instance segmentation
    Zeng, Jiexian
    Ouyang, Huan
    Liu, Min
    Leng, Lu
    Fu, Xiang
    JOURNAL OF KING SAUD UNIVERSITY-COMPUTER AND INFORMATION SCIENCES, 2022, 34 (10) : 9419 - 9427
  • [22] SkiMap plus plus : Real-Time Mapping and Object Recognition for Robotics
    De Gregorio, Daniele
    Cavallari, Tommaso
    Di Stefano, Luigi
    2017 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION WORKSHOPS (ICCVW 2017), 2017, : 660 - 668
  • [23] HTC plus for SAR Ship Instance Segmentation
    Zhang, Tianwen
    Zhang, Xiaoling
    REMOTE SENSING, 2022, 14 (10)
  • [24] FoodMask: Real-time food instance counting, segmentation and recognition
    Nguyen, Huu-Thanh
    Cao, Yu
    Ngo, Chong-Wah
    Chan, Wing-Kwong
    PATTERN RECOGNITION, 2024, 146
  • [25] Real-Time Instance Segmentation Tracking Algorithm in Mixed Reality
    Yu, Dengsha
    Yan, Zifei
    Ming, Baolin
    2021 IEEE 7TH INTERNATIONAL CONFERENCE ON VIRTUAL REALITY (ICVR 2021), 2021, : 368 - 372
  • [26] Edge Assisted Real-time Instance Segmentation on Mobile Devices
    Zhang, Jialin
    Huang, Xiang
    Xu, Jingao
    Wu, Yue
    Ma, Qiang
    Miao, Xin
    Zhang, Li
    Chen, Pengpeng
    Yang, Zheng
    2022 IEEE 42ND INTERNATIONAL CONFERENCE ON DISTRIBUTED COMPUTING SYSTEMS (ICDCS 2022), 2022, : 537 - 547
  • [27] Real-Time Instance Segmentation Method Based on Location Attention
    Liu, Li
    Kong, Yuqi
    KSII TRANSACTIONS ON INTERNET AND INFORMATION SYSTEMS, 2024, 18 (09): : 2483 - 2494
  • [28] Real-Time Instance Segmentation of Traffic Videos for Embedded Devices
    Panero Martinez, Ruben
    Schiopu, Ionut
    Cornelis, Bruno
    Munteanu, Adrian
    SENSORS, 2021, 21 (01) : 1 - 19
  • [29] PolypSeg plus : A Lightweight Context-Aware Network for Real-Time Polyp Segmentation
    Wu, Huisi
    Zhao, Zebin
    Zhong, Jiafu
    Wang, Wei
    Wen, Zhenkun
    Qin, Jing
    IEEE TRANSACTIONS ON CYBERNETICS, 2023, 53 (04) : 2610 - 2621
  • [30] Real-Time Instance Segmentation Models for Identification of Vehicle Parts
    Aldawsari, Abdulmalik
    Yusuf, Syed Adnan
    Souissi, Riad
    AL-Qurishi, Muhammad
    COMPLEXITY, 2023, 2023