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)