Method list_instances

embodiedscan/explorer.py:185–208

List all the instance annotations in the scene. Args: scene (str): Scene name. Returns: list[dict] or None: List of all the instance annotations. If there is no instances, we will return None.

(self, scene)

        return None

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

        Args:
            scene (str): Scene name.

        Returns:
            list[dict] or None: List of all the instance annotations. If there
            is no instances, we will return None.
        """
        for sample in self.data:
            if sample['sample_idx'] == scene:
                res = []
                for instance in sample['instances']:
                    label = self.classes[self.id_to_index[
                        instance['bbox_label_3d']]]
                    res.append({
                        '9dof_bbox': instance['bbox_3d'],
                        'label': label
                    })
                return res

        print('No such scene')
        return None

    def scene_info(self, scene_name):
        """Show the info of the given scene.