About CropSplit and CropSplitGt in mmdet.ops

[ztt0821]

Hi, When I use your code, I found one question about CropSplit and CropSplitGt in mmdet.ops. May I ask the function of the CropSplit. I found that you define self.crop_cuda = CropSplit(2) and self.crop_gt_cuda = CropSplitGt(2) at first, so what's the meaning of "2"? I have print the input feature map size of pred_masks = self.crop_cuda(pred_masks, bbox_dt), which is "[4,288,288, number of coefficient]", and output size is [288,288,number of coefficient]. May I ask what have done in CropSplit?

[ztt0821]

I also found the input size of gt_mask_crop = self.crop_gt_cuda(gt_mask_new, bbox_dt) is [288, 288, number of coefficient], so does self.crop_cuda merge the 4 parts ?

[JialeCao001]

@ztt0821 Yeah. These functions aim at combining 4 parts in cuda.

[ztt0821]

@ztt0821 Yeah. These functions aim at combining 4 parts in cuda.

Thanks for your reply. May I ask what is the meaning of "2" in self.crop_cuda = CropSplit(2)? Does pred_masks = self.crop_cuda(pred_masks, bbox_dt) also crop the segmentation mask based on box location?

[JialeCao001]

@ztt0821 It demonstrates that the regions of detected bounding box are split into 2x2 sub-regions for mask prediction. Yes, it crops the segmentation mask.