Ejemplos de DatasetBuilder.build_panoptic_dataset en Python

Ejemplo n.º 1

Archivo: panoptic_dataset_test.py Proyecto: xdy74/PPLP

    def get_fake_dataset(self, data_split, directory):
        dataset_config = DatasetBuilder.copy_config(
            DatasetBuilder.PANOPTIC_UNITTEST)

        # Overwrite config values
        dataset_config.data_split = data_split
        dataset_config.dataset_dir = directory

        dataset = DatasetBuilder.build_panoptic_dataset(dataset_config)

        return dataset

Ejemplo n.º 2

Archivo: panoptic_utils_test.py Proyecto: xdy74/PPLP

    def setUpClass(cls):
        dataset_config = DatasetBuilder.copy_config(
            DatasetBuilder.PANOPTIC_UNITTEST)

        cls.dataset = DatasetBuilder.build_panoptic_dataset(dataset_config)
        cls.label_dir = cls.dataset.label_dir