论文成果
当前位置: 王兴刚-华中科技大学教师主页 >> 科学研究 >> 论文成果Mask Scoring R-CNN
点击次数:
论文类型:会议论文
第一作者:Huang,Huang,Zhaojin
通讯作者:Wang,Xinggang
合写作者:Huang,Chang,Gong,Yongchao,Lichao
发表刊物:2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)
发表时间:2019-06-15
摘要:Letting a deep network be aware of the quality of its own predictions is an interesting yet important problem. In the task of instance segmentation, the confidence of instance classification is used as mask quality score in most instance segmentation frameworks. However, the mask quality, quantified as the IoU between the instance mask and its ground truth, is usually not well correlated with classification score. In this paper, we study this problem and propose Mask Scoring R-CNN which contains a network block to learn the quality of the predicted instance masks. The proposed network block takes the instance feature and the corresponding predicted mask together to regress the mask IoU. The mask scoring strategy calibrates the misalignment between mask quality and mask score, and improves instance segmentation performance by prioritizing more accurate mask predictions during COCO AP evaluation. By extensive evaluations on the COCO dataset, Mask Scoring R-CNN brings consistent and noticeable gain with different models and outperforms the state-of-the-art Mask R-CNN. We hope our simple and effective approach will provide a new direction for improving instance segmentation. The source code of our method is available at https://github.com/zjhuang22/maskscoring_rcnn.