Function _load_single_frame

demo_render/rgbd_render/data/loader.py:31–34

Load one per-frame npz, keeping only keys the renderer needs.

(path: str)

def _load_single_frame(path: str) -> Dict[str, np.ndarray]:
    """Load one per-frame npz, keeping only keys the renderer needs."""
    data = np.load(path, allow_pickle=False)
    return {key: data[key] for key in data.files if key in _NEEDED_KEYS}


def _load_perframe_dir(dir_path: str, num_workers: int = 16) -> Dict[str, np.ndarray]: