Segmentation label map

[rose-jinyang]

Hello
How are you?
Thanks for contributing to this project.
I want to get one segmentation map containing all the class labels rather than binary mask for each instance.
How can I get it?

[praeclarumjj3]

Hi, you can run a loop to aggregate all the predicted binary masks:

# create an all-zeros mask
single_channel_mask = torch.zeros_like(image)

# loop through all instance masks
for cls_id, mask in zip(result.pred_classes, result.pred_masks):
    mask *= cls_id
    single_channel_mask = torch.max(single_channel_mask, mask)

[rose-jinyang]

Thanks for your help.