INNER CODE UNIT · Python
_load_all
allenai/ai2thor · ai2thor/server.py:47
def _load_all(self):
pass
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"]