search

SamsungLabs / iterdet

[S+SSPR2020] IterDet: Iterative Scheme for Object Detection in Crowded Environments
https://arxiv.org/abs/2005.05708
Mozilla Public License 2.0
210 stars 39 forks source link

IterDet for Instance Segmentation #9

Closed sehyungp92 closed 4 years ago

sehyungp92 commented 4 years ago

Firstly, many thanks for sharing this interesting work.

I was just wondering how this framework could be used or adapted for instance segmentation, and whether you had explored such considerations.

Kind regards, Se

filaPro commented 4 years ago

Hi @sehyungp92

We didn't explored instance segmentation in this work, but mmdetection supports instance segmentation with its MaskRCNN class. Our IterDetFasterRCNN class if just a few line modification of the TwoStageDetector. As MaskRCNN is also inherited from TwoStageDetector it will be easy to modify it to IterDetMaskRCNN by just copying everything from IterDetFasterRCNN and more careful working with mask in simple_test method.