Method list_cameras

embodiedscan/explorer.py:152–183

List all the camera frames in the scene. Args: scene (str): Scene name. Returns: list[str] or None: List of all the frame names. If there is no frames, we will return None.

(self, scene)

        return res

    def list_cameras(self, scene):
        """List all the camera frames in the scene.

        Args:
            scene (str): Scene name.

        Returns:
            list[str] or None: List of all the frame names. If there is no
            frames, we will return None.
        """
        for sample in self.data:
            if sample['sample_idx'] == scene:
                res = []
                dataset = sample['dataset']
                for img in sample['images']:
                    img_path = img['img_path']
                    if dataset == 'scannet':
                        cam_name = img_path.split('/')[-1][:-4]
                    elif dataset == '3rscan':
                        cam_name = img_path.split('/')[-1][:-10]
                    elif dataset == 'matterport3d':
                        cam_name = img_path.split(
                            '/')[-1][:-8] + img_path.split('/')[-1][-7:-4]
                    elif dataset == 'arkitscenes':
                        cam_name = img_path.split('/')[-1][:-4]
                    else:
                        cam_name = img_path.split('/')[-1][:-4]
                    res.append(cam_name)
                return res

        print('No such scene')
        return None

    def list_instances(self, scene):
        """List all the instance annotations in the scene.