boxlevelset icon indicating copy to clipboard operation
boxlevelset copied to clipboard

Instance Segmentation on the Whole Image

Open zhaoyangwei123 opened this issue 2 years ago • 1 comments

@LiWentomng In your paper, you mentioned restricting the level set to gt bounding box for segmentation and also mentioned that the training result is poor on the whole image. How can I get the boxlevelset to run on the whole image?

zhaoyangwei123 avatar Jan 04 '23 05:01 zhaoyangwei123

Hello, @zhaoyangwei123 The gt bounding box can prevent the interference from background and other objects. We achieve this goal by the binary box-based mask (box_mask_target) as following,

           mask_scores_phi = mask_scores_concat * box_mask_target
           img_target_wbox = img_target * box_mask_target
           high_feature = torch.cat((deep_stru_feature_img, deep_stru_feature_lst), dim=1) * box_mask_target

The region outside gt box for level set is zero, which are not participate in optimization.

You can remove the box_mask_target to run on the whole image.

LiWentomng avatar Jan 06 '23 05:01 LiWentomng