def setup(self): tf_records_path = os.path.join(self.export_dir, "tf.records") self._filename_maker = fou.UniqueFilenameMaker() self._example_generator = self._make_example_generator() self._tf_records_writer = TFRecordsWriter(tf_records_path, num_shards=self.num_shards) self._tf_records_writer.__enter__()
def setup(self): self._data_dir = os.path.join(self.export_dir, "data") self._labels_path = os.path.join(self.export_dir, "labels.json") self._annotations = [] self._filename_maker = fou.UniqueFilenameMaker( output_dir=self._data_dir, default_ext=self.image_format )
def setup(self): self._data_dir = os.path.join(self.export_dir, "data") self._frame_labels_dir = os.path.join(self.export_dir, "frames") self._metadata_path = os.path.join(self.export_dir, "metadata.json") self._samples_path = os.path.join(self.export_dir, "samples.json") self._metadata = {} self._samples = [] self._filename_maker = fou.UniqueFilenameMaker( output_dir=self._data_dir)
def setup(self): self._data_dir = os.path.join(self.export_dir, "data") self._labels_path = os.path.join(self.export_dir, "labels.json") self._labels_dict = {} self._filename_maker = fou.UniqueFilenameMaker( output_dir=self._data_dir, default_ext=self.image_format, ignore_exts=True, )
def setup(self): self._labeled_dataset = etad.LabeledVideoDataset.create_empty_dataset( self.export_dir) self._data_dir = self._labeled_dataset.data_dir self._labels_dir = self._labeled_dataset.labels_dir self._filename_maker = fou.UniqueFilenameMaker( output_dir=self._data_dir, ignore_exts=True, )
def setup(self): self._data_dir = os.path.join(self.export_dir, "data") self._labels_dir = os.path.join(self.export_dir, "labels") self._filename_maker = fou.UniqueFilenameMaker( output_dir=self._data_dir ) self._writer = CVATVideoAnnotationWriter() etau.ensure_dir(self._data_dir) etau.ensure_dir(self._labels_dir)
def setup(self): self._labeled_dataset = etad.LabeledImageDataset.create_empty_dataset( self.export_dir) self._data_dir = self._labeled_dataset.data_dir self._labels_dir = self._labeled_dataset.labels_dir self._filename_maker = fou.UniqueFilenameMaker( output_dir=self._data_dir, default_ext=self.image_format, ignore_exts=True, )
def setup(self): self._data_dir = os.path.join(self.export_dir, "data") self._labels_dir = os.path.join(self.export_dir, "labels") self._filename_maker = fou.UniqueFilenameMaker( output_dir=self._data_dir, default_ext=self.image_format, ignore_exts=True, ) self._writer = KITTIAnnotationWriter() etau.ensure_dir(self._data_dir) etau.ensure_dir(self._labels_dir)
def setup(self): self._data_dir = os.path.join(self.export_dir, "data") self._labels_path = os.path.join(self.export_dir, "labels.json") self._image_id = -1 self._anno_id = -1 self._images = [] self._annotations = [] self._classes = set() self._filename_maker = fou.UniqueFilenameMaker( output_dir=self._data_dir, default_ext=self.image_format) self._has_labels = False self._parse_classes()
def setup(self): if self.classes is not None: self._labels_map_rev = _to_labels_map_rev(self.classes) else: self._labels_map_rev = None self._data_dir = os.path.join(self.export_dir, "data") self._labels_path = os.path.join(self.export_dir, "labels.json") self._image_id = -1 self._anno_id = -1 self._images = [] self._annotations = [] self._classes = set() self._filename_maker = fou.UniqueFilenameMaker( output_dir=self._data_dir, default_ext=self.image_format )
def setup(self): self._obj_names_path = os.path.join(self.export_dir, "obj.names") self._images_path = os.path.join(self.export_dir, "images.txt") self._data_dir = os.path.join(self.export_dir, "data") self._classes = {} self._labels_map_rev = {} self._images = [] self._filename_maker = fou.UniqueFilenameMaker( output_dir=self._data_dir, default_ext=self.image_format, ignore_exts=True, ) self._writer = YOLOAnnotationWriter() etau.ensure_dir(self._data_dir) self._parse_classes()
def draw_labeled_images(samples, label_fields, anno_dir, annotation_config=None): """Renders annotated versions of the image samples with label field(s) overlaid to the given directory. The filenames of the sample images are maintained, unless a name conflict would occur in ``anno_dir``, in which case an index of the form ``"-%d" % count`` is appended to the base filename. The images are written in format ``fo.config.default_image_ext``. Args: samples: an iterable of :class:`fiftyone.core.sample.Sample` instances label_fields: the list of :class:`fiftyone.core.labels.ImageLabel` fields to render anno_dir: the directory to write the annotated images annotation_config (None): an :class:`AnnotationConfig` specifying how to render the annotations Returns: the list of paths to the labeled images """ if annotation_config is None: annotation_config = _DEFAULT_ANNOTATION_CONFIG filename_maker = fou.UniqueFilenameMaker(output_dir=anno_dir) output_ext = fo.config.default_image_ext outpaths = [] with fou.ProgressBar() as pb: for sample in pb(samples): outpath = filename_maker.get_output_path(sample.filepath, output_ext=output_ext) draw_labeled_image( sample, label_fields, outpath, annotation_config=annotation_config, ) outpaths.append(outpath) return outpaths
def _setup(self): self._filename_maker = fou.UniqueFilenameMaker( output_dir=self.dataset_dir, default_ext=self.image_format )
def setup(self): self._filename_maker = fou.UniqueFilenameMaker( output_dir=self.export_dir)
def setup(self): self._data_dir = os.path.join(self.export_dir, "data") self._samples_path = os.path.join(self.export_dir, "samples.json") self._samples = [] self._filename_maker = fou.UniqueFilenameMaker( output_dir=self._data_dir)
def _setup(self): self._filename_maker = fou.UniqueFilenameMaker( output_dir=self.dataset_dir)