module documentation
Utilities for working with datasets in Berkeley DeepDrive (BDD) format.
Class |
|
Exporter that writes BDD datasets to disk. |
Class |
|
Importer for BDD datasets stored on disk. |
Function | load |
Loads the BDD annotations from the given JSON file. |
Function | parse |
Parses the raw BDD100K download files in the specified directory into per-split directories in BDD format. |
Variable | logger |
Undocumented |
Function | _detection |
Undocumented |
Function | _ensure |
Undocumented |
Function | _ensure |
Undocumented |
Function | _ensure |
Undocumented |
Function | _filter |
Undocumented |
Function | _get |
Undocumented |
Function | _make |
Undocumented |
Function | _parse |
Undocumented |
Function | _parse |
Undocumented |
Function | _parse |
Undocumented |
Function | _parse |
Undocumented |
Function | _polyline |
Undocumented |
Function | _raise |
Undocumented |
Loads the BDD annotations from the given JSON file.
See :ref:`this page <BDDDataset-import>` for format details.
Parameters | |
json | the path to the annotations JSON file |
Returns | |
a dict mapping filenames to BDD annotation dicts |
Parses the raw BDD100K download files in the specified directory into per-split directories in BDD format.
This function assumes that the input source_dir contains the following contents:
source_dir/ labels/ bdd100k_labels_images_train.json bdd100k_labels_images_val.json images/ 100k/ train/ test/ val/ ...
and will populate dataset_dir as follows:
dataset_dir/ train/ data/ labels.json validation/ data/ labels.json test/ data/
Parameters | |
source | the source directory containing the manually downloaded BDD100K files |
dataset | the directory to construct the output split directories |
copyTrue | whether to move (False) or create copies (True) of the source files when populating dataset_dir |
overwrite:False | whether to overwrite existing files/directories in the output location, if they exist |
Raises | |
OSError | if any required source files are not present |