Detectron 2 pred_masks checking to see if a given class has been detected?

vision
1

How can I check to see if the pred_masks is false for a given COCO class (e.g. bicycle)?

(Pdb) instances.pred_masks
tensor([[[False, False, False,  ..., False, False, False],
         [False, False, False,  ..., False, False, False],
         [False, False, False,  ..., False, False, False],
         ...,
         [False, False, False,  ..., False, False, False],
         [False, False, False,  ..., False, False, False],
         [False, False, False,  ..., False, False, False]],

        [[False, False, False,  ..., False, False, False],
         [False, False, False,  ..., False, False, False],
         [False, False, False,  ..., False, False, False],
         ...,
         [False, False, False,  ..., False, False, False],
         [False, False, False,  ..., False, False, False],
         [False, False, False,  ..., False, False, False]]], device='cuda:0')
(Pdb) torch.all(instances.pred_masks == False)
tensor(False, device='cuda:0')

From the above it shows it is True for some classes, right? How can I check if it is true for a class name of my own interest?

I also have this dimension and am not exactly sure what it represents:

(Pdb) instances.pred_masks.shape
torch.Size([2, 414, 640])

[ a double-post of checking the pred_masks against COCO classes of interest · Discussion #3036 · facebookresearch/detectron2 · GitHub ]

2

if coco_metadata.thing_classes.index(args.class_name) in instances.pred_classes: