def from_images_dir(images_dir, recursive=True, num_parallel_calls=None):
"""Creates a ``tf.data.Dataset`` for the given directory of images.
Args:
images_dir: a directory of images
recursive (True): whether to recursively traverse subdirectories
num_parallel_calls (None): the number of samples to read
asynchronously in parallel. See
https://www.tensorflow.org/api_docs/python/tf/data/Dataset#map for
details
Returns:
a ``tf.data.Dataset`` that emits decoded images
"""
image_paths = foud.parse_images_dir(images_dir, recursive=recursive)
return from_images(image_paths, num_parallel_calls=num_parallel_calls)
def _make_images_list(images_dir):
logger.info("Computing image metadata for '%s'", images_dir)
image_paths = foud.parse_images_dir(images_dir)
images = []
with fou.ProgressBar() as pb:
for idx, image_path in pb(enumerate(image_paths)):
metadata = fom.ImageMetadata.build_for(image_path)
images.append({
"id": idx,
"file_name": os.path.basename(image_path),
"height": metadata.height,
"width": metadata.width,
"license": None,
"coco_url": None,
})
return images
def add_images_dir(self, images_dir, tags=None, recursive=True):
"""Adds the given directory of images to the dataset.
See :class:`fiftyone.types.dataset_types.ImageDirectory` for format
details. In particular, note that files with non-image MIME types are
omitted.
This operation does not read the images.
Args:
images_dir: a directory of images
tags (None): an optional list of tags to attach to each sample
recursive (True): whether to recursively traverse subdirectories
Returns:
a list of IDs of the samples in the dataset
"""
image_paths = foud.parse_images_dir(images_dir, recursive=recursive)
sample_parser = foud.ImageSampleParser()
return self.add_images(image_paths, sample_parser, tags=tags)