mmrotate icon indicating copy to clipboard operation
mmrotate copied to clipboard

Support the instance segmentation task in projects

Open zytx121 opened this issue 2 years ago • 5 comments

This project implements a Mask RCNN for rotated boxes. Benefiting from the BoxType design, we only need to modify the code slightly in mmrotate, and we can support the instance segmentation task. Although the rotated box indicator will drop slightly after adding the mask head, it may help improve the instance segmentation task.

P0262__1024__824___1414 P0003__1024__123___0

We hope this project can inspire you and welcome you to explore more uses of mmrotate!

zytx121 avatar Feb 15 '23 03:02 zytx121

Thanks for this contribution, I believe it's very valuable to expand the libraries functionality! One suggestion for an improvement of the mask delineation: Replace the proposed matching of rotated ROI aligned feature maps with hbox mask targets (using rbox2hox in mask_target.py -> mask_target_single & rotated_fcn_mask_head.py -> _do_paste_mask()) with an rotated ROI alignment for the mask targets. Thereby the feature maps as well as the mask targets are actually spatially matching, which allows a better mask prediction. To illustrate the improvement, you can find a comparison of both approaches for the use case of agricultural parcel delineation below.

grafik

fkroeber avatar May 11 '23 13:05 fkroeber

@zytx121

yangxue0827 avatar May 26 '23 07:05 yangxue0827

When I use my own training set, why the box is shifted, but the segm is correct? @zytx121 @fkroeber N01_18

Ataraxi8 avatar Jun 29 '23 09:06 Ataraxi8

@fkroeber Thank you for sharing, the results are much better than now! May I ask if you are willing to provide us with a PR?

zytx121 avatar Jun 30 '23 02:06 zytx121

Hello, I am new to MMRotate. I've found this library while I was searching for an instance segmentation model with oriented bounding boxes. Thank you @zytx121, I think this project will solve my problem. I have seen that DOTA dataset type is privileged in MMRotate. But I guess that masks have to be provided. Therefore, I assume that DOTA dataset is not supported by your project. This is why I ask you what type of dataset have to be used in this project ?

tcotte avatar Jun 05 '24 13:06 tcotte