luxonis_ml.data.exporters.YoloV4Exporter

class documentation

class YoloV4Exporter(BaseExporter):

Constructor: YoloV4Exporter(dataset_identifier, output_path, max_partition_size_gb)

View In Hierarchy

Export object detection data in YOLOv4 format.

Method __init__ Undocumented
Method export Convert the prepared dataset into the exporter's format.
Method get_split_names Undocumented
Method supported_ann_types Return task types supported by this exporter.
Instance Variable class_names Class names ordered by class ID.
Instance Variable class_to_id Class-name to class-ID mapping.
Instance Variable current_size Undocumented
Method _dump_annotations Undocumented
Method _dump_annotations_v4 Undocumented
Method _get_data_path Return the folder path to store data files for this split.
Method _maybe_roll_partition Undocumented
Method _yaml_filename Undocumented

Inherited from BaseExporter:

Instance Variable dataset_identifier Name or identifier used for exported paths.
Instance Variable image_indices Per-image export indices used by concrete exporters.
Instance Variable max_partition_size Optional maximum partition size in bytes.
Instance Variable max_partition_size_gb Optional maximum partition size in GiB.
Instance Variable output_path Directory where the export is written.
Instance Variable part Current partition index, or None when partitioning is disabled.
def __init__(self, dataset_identifier: str, output_path: Path, max_partition_size_gb: float | None):
overrides luxonis_ml.data.exporters.BaseExporter.__init__

Undocumented

def export(self, prepared_ldf: PreparedLDF):
overrides luxonis_ml.data.exporters.BaseExporter.export

Convert the prepared dataset into the exporter's format.

Parameters
prepared_ldf:PreparedLDFDataset data prepared for export.
Raises
NotImplementedErrorAlways raised by the abstract base implementation.
def get_split_names(self) -> dict[str, str]:

Undocumented

def supported_ann_types(self) -> list[str]:
overrides luxonis_ml.data.exporters.BaseExporter.supported_ann_types

Return task types supported by this exporter.

Returns
list[str]Supported annotation task types.
Raises
NotImplementedErrorAlways raised by the abstract base implementation.
class_names: list[str] =

Class names ordered by class ID.

class_to_id: dict[str, int] =

Class-name to class-ID mapping.

current_size: int =
overrides luxonis_ml.data.exporters.BaseExporter.current_size

Undocumented

def _dump_annotations(self, annotation_splits: dict[str, dict[str, list[BBox]]], output_path: Path, part: int | None = None):
overrides luxonis_ml.data.exporters.BaseExporter._dump_annotations

Undocumented

def _dump_annotations_v4(self, annotation_splits: dict[str, dict[str, list[BBox]]], output_path: Path, part: int | None = None):

Undocumented

def _get_data_path(self, output_path: Path, split: str, part: int | None = None) -> Path:
overrides luxonis_ml.data.exporters.BaseExporter._get_data_path

Return the folder path to store data files for this split.

def _maybe_roll_partition(self, annotation_splits: dict[str, dict[str, list[BBox]]], additional_size: int) -> dict[str, dict[str, list[BBox]]]:

Undocumented

def _yaml_filename(self) -> str:

Undocumented