Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Hybrid Task Cascade for Instance Segmentation

About

Cascade is a classic yet powerful architecture that has boosted performance on various tasks. However, how to introduce cascade to instance segmentation remains an open question. A simple combination of Cascade R-CNN and Mask R-CNN only brings limited gain. In exploring a more effective approach, we find that the key to a successful instance segmentation cascade is to fully leverage the reciprocal relationship between detection and segmentation. In this work, we propose a new framework, Hybrid Task Cascade (HTC), which differs in two important aspects: (1) instead of performing cascaded refinement on these two tasks separately, it interweaves them for a joint multi-stage processing; (2) it adopts a fully convolutional branch to provide spatial context, which can help distinguishing hard foreground from cluttered background. Overall, this framework can learn more discriminative features progressively while integrating complementary features together in each stage. Without bells and whistles, a single HTC obtains 38.4 and 1.5 improvement over a strong Cascade Mask R-CNN baseline on MSCOCO dataset. Moreover, our overall system achieves 48.6 mask AP on the test-challenge split, ranking 1st in the COCO 2018 Challenge Object Detection Task. Code is available at: https://github.com/open-mmlab/mmdetection.

Kai Chen, Jiangmiao Pang, Jiaqi Wang, Yu Xiong, Xiaoxiao Li, Shuyang Sun, Wansen Feng, Ziwei Liu, Jianping Shi, Wanli Ouyang, Chen Change Loy, Dahua Lin• 2019

Related benchmarks

TaskDatasetResultRank
Object DetectionCOCO 2017 (val)
AP62.5
2643
Object DetectionCOCO (test-dev)
mAP63.1
1239
Instance SegmentationCOCO 2017 (val)
APm0.422
1201
Object DetectionCOCO (val)
mAP44.7
633
Object DetectionCOCO v2017 (test-dev)
mAP57.7
499
Instance SegmentationCOCO (val)
APmk42.5
475
Instance SegmentationCOCO (test-dev)
APM44.2
380
Instance SegmentationCOCO 2017 (test-dev)
AP (Overall)49
253
Camouflaged Object DetectionCOD10K (test)
S-measure (S_alpha)0.548
224
Object DetectionMS-COCO (val)
mAP0.432
211
Showing 10 of 50 rows

Other info

Code

Follow for update