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)
