Attend the Women in ML Symposium on December 7 Register now

tfm.vision.maskrcnn.zero_out_disallowed_class_ids

Zero out IDs of classes not in allowed_class_ids.

batch_class_ids A [batch_size, num_instances] int tensor of input class IDs.
allowed_class_ids A python list of class IDs which we want to allow.

filtered_class_ids A [batch_size, num_instances] int tensor with any class ID not in allowed_class_ids set to 0.