INNER CODE UNIT · Python

LazyClassSegmentationMasks

allenai/ai2thor · ai2thor/server.py:157

class LazyClassSegmentationMasks(LazyMask):
    def __init__(self, instance_masks: LazyInstanceSegmentationMasks):
        self.instance_masks = instance_masks
        self._loaded = False
        self._masks: Dict[str, np.ndarray] = {}

    def _load_all(self):
        if not self._loaded:
            all_integer_keys = self.instance_masks.unique_integer_keys
            for cls, colors in self.instance_masks.class_colors.items():
                for color in colors:
                    if self.instance_masks._integer_color_key(color) in all_integer_keys:
                        self.__getitem__(cls)
                        break
        self._loaded = True

    def mask(self, key: str, default: Optional[np.ndarray] = None) -> Optional[np.ndarray]:
        if key in self._masks:

View source record →