def setup(self):
self._sample_parser = FiftyOneVideoLabelsSampleParser(
expand=self.expand,
prefix=self.prefix,
labels_dict=self.labels_dict,
multilabel=self.multilabel,
skip_non_categorical=self.skip_non_categorical,
)
self._labeled_dataset = etads.load_dataset(self.dataset_dir)
self._description = self._labeled_dataset.dataset_index.description
self._num_samples = len(self._labeled_dataset)
if self.max_samples is not None:
self._num_samples = min(self._num_samples, self.max_samples)
def get_num_samples(dataset_dir):
return len(etads.load_dataset(dataset_dir))
def setup(self):
self._sample_parser = FiftyOneImageLabelsSampleParser()
self._labeled_dataset = etads.load_dataset(self.dataset_dir)