INNER CODE UNIT · Python

background_keys

allenai/ai2thor · ai2thor/server.py:191

            background_keys = all_integer_keys - metadata_color_keys
            for ik in background_keys:
                mask = self.instance_masks.instance_segmentation_frame_uint32 == ik
                class_mask = np.logical_or(class_mask, mask)

        elif "|" not in key:
            for color in self.instance_masks.class_colors.get(key, []):
                mask = (
                    self.instance_masks.instance_segmentation_frame_uint32
                    == self.instance_masks._integer_color_key(color)
                )
                class_mask = np.logical_or(class_mask, mask)

        if class_mask.any():
            self._masks[key] = class_mask
            return class_mask
        else:
            return default

View source record →