fiftyone.utils.voc#
Utilities for working with datasets in VOC format.
Classes:
|
Importer for VOC detection datasets stored on disk. |
|
Exporter that writes VOC detection datasets to disk. |
|
Class representing a VOC annotations file. |
|
An object in VOC detection format. |
|
A bounding box in VOC detection format. |
Class for writing annotations in VOC format. |
Functions:
|
Loads the VOC detection annotations from the given XML file. |
- class fiftyone.utils.voc.VOCDetectionDatasetImporter(dataset_dir=None, data_path=None, labels_path=None, include_all_data=False, extra_attrs=True, shuffle=False, seed=None, max_samples=None)#
Bases:
LabeledImageDatasetImporter,ImportPathsMixinImporter for VOC detection datasets stored on disk.
See this page for format details.
- Parameters:
dataset_dir (None) – the dataset directory. If omitted,
data_pathand/orlabels_pathmust be provideddata_path (None) –
an optional parameter that enables explicit control over the location of the media. Can be any of the following:
a folder name like
"data"or"data/"specifying a subfolder ofdataset_dirwhere the media files residean absolute directory path where the media files reside. In this case, the
dataset_dirhas no effect on the location of the dataa filename like
"data.json"specifying the filename of the JSON data manifest file indataset_diran absolute filepath specifying the location of the JSON data manifest. In this case,
dataset_dirhas no effect on the location of the dataa dict mapping filenames to absolute filepaths
If None, this parameter will default to whichever of
data/ordata.jsonexists in the dataset directorylabels_path (None) –
an optional parameter that enables explicit control over the location of the labels. Can be any of the following:
a folder name like
"labels"or"labels/"specifying the location of the labels indataset_diran absolute folder path to the labels. In this case,
dataset_dirhas no effect on the location of the labels
If None, the parameter will default to
labels/include_all_data (False) – whether to generate samples for all images in the data directory (True) rather than only creating samples for images with label entries (False)
extra_attrs (True) –
whether to load extra annotation attributes onto the imported labels. Supported values are:
True: load all extra attributes foundFalse: do not load extra attributesa name or list of names of specific attributes to load
shuffle (False) – whether to randomly shuffle the order in which the samples are imported
seed (None) – a random seed to use when shuffling
max_samples (None) – a maximum number of samples to import. By default, all samples are imported
Attributes:
Whether this importer produces a dataset info dictionary.
Whether this importer produces
fiftyone.core.metadata.ImageMetadatainstances for each image.The
fiftyone.core.labels.Labelclass(es) returned by this importer.Methods:
setup()Performs any necessary setup before importing the first sample in the dataset.
close(*args)Performs any necessary actions after the last sample has been imported.
Returns the dataset info for the dataset.
- property has_dataset_info#
Whether this importer produces a dataset info dictionary.
- property has_image_metadata#
Whether this importer produces
fiftyone.core.metadata.ImageMetadatainstances for each image.
- property label_cls#
The
fiftyone.core.labels.Labelclass(es) returned by this importer.This can be any of the following:
a
fiftyone.core.labels.Labelclass. In this case, the importer is guaranteed to return labels of this typea list or tuple of
fiftyone.core.labels.Labelclasses. In this case, the importer can produce a single label field of any of these typesa dict mapping keys to
fiftyone.core.labels.Labelclasses. In this case, the importer will return label dictionaries with keys and value-types specified by this dictionary. Not all keys need be present in the imported labelsNone. In this case, the importer makes no guarantees about the labels that it may return
- setup()#
Performs any necessary setup before importing the first sample in the dataset.
This method is called when the importer’s context manager interface is entered,
DatasetImporter.__enter__().
- close(*args)#
Performs any necessary actions after the last sample has been imported.
This method is called when the importer’s context manager interface is exited,
DatasetImporter.__exit__().- Parameters:
*args – the arguments to
DatasetImporter.__exit__()
- get_dataset_info()#
Returns the dataset info for the dataset.
By convention, this method should be called after all samples in the dataset have been imported.
- Returns:
a dict of dataset info
- class fiftyone.utils.voc.VOCDetectionDatasetExporter(export_dir=None, data_path=None, labels_path=None, export_media=None, rel_dir=None, include_paths=True, image_format=None, extra_attrs=True)#
Bases:
LabeledImageDatasetExporter,ExportPathsMixinExporter that writes VOC detection datasets to disk.
See this page for format details.
- Parameters:
export_dir (None) – the directory to write the export. This has no effect if
data_pathandlabels_pathare absolute pathsdata_path (None) –
an optional parameter that enables explicit control over the location of the exported media. Can be any of the following:
a folder name like
"data"or"data/"specifying a subfolder ofexport_dirin which to export the mediaan absolute directory path in which to export the media. In this case, the
export_dirhas no effect on the location of the dataa JSON filename like
"data.json"specifying the filename of the manifest file inexport_dirgenerated whenexport_mediais"manifest"an absolute filepath specifying the location to write the JSON manifest file when
export_mediais"manifest". In this case,export_dirhas no effect on the location of the data
If None, the default value of this parameter will be chosen based on the value of the
export_mediaparameterlabels_path (None) –
an optional parameter that enables explicit control over the location of the exported labels. Can be any of the following:
a folder name like
"labels"or"labels/"specifying the location inexport_dirin which to export the labelsan absolute folder path to which to export the labels. In this case, the
export_dirhas no effect on the location of the labels
If None, the labels will be exported into
export_dirusing the default folder nameexport_media (None) –
controls how to export the raw media. The supported values are:
True: copy all media files into the output directoryFalse: don’t export media"move": move all media files into the output directory"symlink": create symlinks to the media files in the output directory"manifest": create adata.jsonin the output directory that maps UUIDs used in the labels files to the filepaths of the source media, rather than exporting the actual media
If None, the default value of this parameter will be chosen based on the value of the
data_pathparameterrel_dir (None) – an optional relative directory to strip from each input filepath to generate a unique identifier for each image. When exporting media, this identifier is joined with
data_pathandlabels_pathto generate output paths for each exported image and labels file. This argument allows for populating nested subdirectories that match the shape of the input paths. The path is converted to an absolute path (if necessary) viafiftyone.core.storage.normalize_path()include_paths (True) – whether to include the absolute paths to the images in the
<path>elements of the exported XMLimage_format (None) – the image format to use when writing in-memory images to disk. By default,
fiftyone.config.default_image_extis usedextra_attrs (True) –
whether to include extra object attributes in the exported labels. Supported values are:
True: export all extra attributes foundFalse: do not export extra attributesa name or list of names of specific attributes to export
Attributes:
Whether this exporter requires
fiftyone.core.metadata.ImageMetadatainstances for each sample being exported.The
fiftyone.core.labels.Labelclass(es) exported by this exporter.Methods:
setup()Performs any necessary setup before exporting the first sample in the dataset.
export_sample(image_or_path, detections[, ...])Exports the given sample to the dataset.
close(*args)Performs any necessary actions after the last sample has been exported.
log_collection(sample_collection)Logs any relevant information about the
fiftyone.core.collections.SampleCollectionwhose samples will be exported.- property requires_image_metadata#
Whether this exporter requires
fiftyone.core.metadata.ImageMetadatainstances for each sample being exported.
- property label_cls#
The
fiftyone.core.labels.Labelclass(es) exported by this exporter.This can be any of the following:
a
fiftyone.core.labels.Labelclass. In this case, the exporter directly exports labels of this typea list or tuple of
fiftyone.core.labels.Labelclasses. In this case, the exporter can export a single label field of any of these typesa dict mapping keys to
fiftyone.core.labels.Labelclasses. In this case, the exporter can handle label dictionaries with value-types specified by this dictionary. Not all keys need be present in the exported label dictsNone. In this case, the exporter makes no guarantees about the labels that it can export
- setup()#
Performs any necessary setup before exporting the first sample in the dataset.
This method is called when the exporter’s context manager interface is entered,
DatasetExporter.__enter__().
- export_sample(image_or_path, detections, metadata=None)#
Exports the given sample to the dataset.
- Parameters:
image_or_path – an image or the path to the image on disk
label – an instance of
label_cls(), or a dictionary mapping field names tofiftyone.core.labels.Labelinstances, orNoneif the sample is unlabeledmetadata (None) – a
fiftyone.core.metadata.ImageMetadatainstance for the sample. Only required whenrequires_image_metadata()isTrue
- close(*args)#
Performs any necessary actions after the last sample has been exported.
This method is called when the exporter’s context manager interface is exited,
DatasetExporter.__exit__().- Parameters:
*args – the arguments to
DatasetExporter.__exit__()
- log_collection(sample_collection)#
Logs any relevant information about the
fiftyone.core.collections.SampleCollectionwhose samples will be exported.Subclasses can optionally implement this method if their export format can record information such as the
fiftyone.core.collections.SampleCollection.info()of the collection being exported.By convention, this method must be optional; i.e., if it is not called before the first call to
export_sample(), then the exporter must make do without any information about thefiftyone.core.collections.SampleCollection(which may not be available, for example, if the samples being exported are not stored in a collection).- Parameters:
sample_collection – the
fiftyone.core.collections.SampleCollectionwhose samples will be exported
- class fiftyone.utils.voc.VOCAnnotation(path=None, folder=None, filename=None, segmented=None, metadata=None, objects=None)#
Bases:
objectClass representing a VOC annotations file.
- Parameters:
path (None) – the path to the image
folder (None) – the name of the folder containing the image
filename (None) – the image filename
segmented (None) – whether the objects are segmented
metadata (None) – a
fiftyone.core.metadata.ImageMetadatainstanceobjects (None) – a list of
VOCObjectinstances
Methods:
to_detections([extra_attrs])Returns a
fiftyone.core.labels.Detectionsrepresentation of the objects in the annotation.from_labeled_image(metadata, detections[, ...])Creates a
VOCAnnotationinstance for the given labeled image data.from_xml(xml_path)Creates a
VOCAnnotationinstance from an XML annotations file.from_dict(d)Creates a
VOCAnnotationinstance from a JSON dict representation.- to_detections(extra_attrs=True)#
Returns a
fiftyone.core.labels.Detectionsrepresentation of the objects in the annotation.- Parameters:
extra_attrs (True) –
whether to load extra annotation attributes onto the imported labels. Supported values are:
True: load all extra attributes foundFalse: do not load extra attributesa name or list of names of specific attributes to load
- Returns:
- classmethod from_labeled_image(metadata, detections, path=None, filename=None, extra_attrs=True)#
Creates a
VOCAnnotationinstance for the given labeled image data.- Parameters:
metadata – a
fiftyone.core.metadata.ImageMetadatainstance for the imagedetections – a
fiftyone.core.labels.Detectionspath (None) – the absolute path to the image
filename (None) – the filename of the image
extra_attrs (True) –
whether to include extra object attributes. Supported values are:
True: include all extra attributes foundFalse: do not include extra attributesa name or list of names of specific attributes to include
- Returns:
- classmethod from_xml(xml_path)#
Creates a
VOCAnnotationinstance from an XML annotations file.- Parameters:
xml_path – the path to the XML file
- Returns:
- classmethod from_dict(d)#
Creates a
VOCAnnotationinstance from a JSON dict representation.- Parameters:
d – a JSON dict
- Returns:
- class fiftyone.utils.voc.VOCObject(name, bndbox, **attributes)#
Bases:
objectAn object in VOC detection format.
- Parameters:
name – the object label
bndbox – a
VOCBoundingBox**attributes – additional custom attributes
Methods:
Creates a
VOCObjectfrom a VOC annotation dict.from_detection(detection, metadata[, ...])Creates a
VOCObjectfrom afiftyone.core.labels.Detection.to_detection(frame_size[, extra_attrs])Returns a
fiftyone.core.labels.Detectionrepresentation of the object.- classmethod from_annotation_dict(d)#
Creates a
VOCObjectfrom a VOC annotation dict.- Parameters:
d – an annotation dict
- Returns:
- classmethod from_detection(detection, metadata, extra_attrs=True)#
Creates a
VOCObjectfrom afiftyone.core.labels.Detection.- Parameters:
detection – a
fiftyone.core.labels.Detectionmetadata – a
fiftyone.core.metadata.ImageMetadatainstance for the imageextra_attrs (True) –
whether to include extra object attributes. Supported values are:
True: include all extra attributes foundFalse: do not include extra attributesa name or list of names of specific attributes to include
- Returns:
- to_detection(frame_size, extra_attrs=True)#
Returns a
fiftyone.core.labels.Detectionrepresentation of the object.- Parameters:
frame_size – the
(width, height)of the imageextra_attrs (True) –
whether to include extra annotation attributes on the object. Supported values are:
True: include all extra attributes foundFalse: do not include extra attributesa name or list of names of specific attributes to include
- Returns:
- class fiftyone.utils.voc.VOCBoundingBox(xmin, ymin, xmax, ymax)#
Bases:
objectA bounding box in VOC detection format.
- Parameters:
xmin – the top-left x coordinate
ymin – the top-left y coordinate
xmax – the bottom-right x coordinate
ymax – the bottom-right y coordinate
Methods:
Creates a
VOCBoundingBoxfrom abndboxdict.from_detection_format(bounding_box, frame_size)Creates a
VOCBoundingBoxfrom a bounding box stored infiftyone.core.labels.Detectionformat.to_detection_format(frame_size)Returns a representation of the bounding box suitable for storing in the
bounding_boxfield of afiftyone.core.labels.Detection.- classmethod from_bndbox_dict(d)#
Creates a
VOCBoundingBoxfrom abndboxdict.- Parameters:
d – a
bndboxdict- Returns:
- classmethod from_detection_format(bounding_box, frame_size)#
Creates a
VOCBoundingBoxfrom a bounding box stored infiftyone.core.labels.Detectionformat.- Parameters:
bounding_box –
[x-top-left, y-top-left, width, height]frame_size – the
(width, height)of the image
- Returns:
- to_detection_format(frame_size)#
Returns a representation of the bounding box suitable for storing in the
bounding_boxfield of afiftyone.core.labels.Detection.- Parameters:
frame_size – the
(width, height)of the image- Returns:
[x-top-left, y-top-left, width, height]
- class fiftyone.utils.voc.VOCAnnotationWriter#
Bases:
objectClass for writing annotations in VOC format.
See this page for format details.
Methods:
write(annotation, xml_path)Writes the annotations to disk.
- write(annotation, xml_path)#
Writes the annotations to disk.
- Parameters:
annotation – a
VOCAnnotationinstancexml_path – the path to write the annotation XML file
- fiftyone.utils.voc.load_voc_detection_annotations(xml_path)#
Loads the VOC detection annotations from the given XML file.
See this page for format details.
- Parameters:
xml_path – the path to the annotations XML file
- Returns:
a
VOCAnnotationinstance