INNER CODE UNIT · Python

__iter__

allenai/ai2thor · ai2thor/server.py:50

    def __iter__(self):
        self._load_all()
        return iter(self._masks)

    def __len__(self):
        self._load_all()
        return len(self._masks)


class LazyInstanceSegmentationMasks(LazyMask):
    def __init__(self, image_ids_data: bytes, metadata: dict):
        self._masks: Dict[str, np.ndarray] = {}
        self._loaded = False
        screen_width = metadata["screenWidth"]
        screen_height = metadata["screenHeight"]
        item_size = int(len(image_ids_data) / (screen_width * screen_height))
        self._unique_integer_keys: Optional[Set[np.uint32]] = None
        self._empty_mask: Optional[np.ndarray] = None

View source record →