torchsig.transforms.target_transforms.YOLOLabel

class torchsig.transforms.target_transforms.YOLOLabel(**kwargs)

Bases: TargetTransform

Adds a YOLO_label to a signal, in the form of a list of tuples (cid, cx, cy, width, height)

output

Structure to aggregate YOLO labels (“dict”, “list”). Defaults to “list”.

Type:

str, optional

Methods

add_parent	Add parent Seedable object and set up RNGs accordingly
get_distribution
get_second_seed	Gets second seed, usually used to seed both torch and numpy generators with slightly different seeds
seed	Seed number generators with given seed.
setup_rngs	Initialize torch and numpy number generators, and update its children.
update	Update bookeeping for signals
update_from_parent	Update numpy and torch number generators with parent seed

Attributes

output_list

__init__(**kwargs)

Transform initialization as Seedable.

__call__(metadatas: List[Dict[str, Any]], enable_verify=True) → List[Any] | Dict[str, Any]

Applies the target transform to a list of signal metadata.

Parameters:

• metadatas (List[Dict[str, Any]]) – The list of metadata dictionaries to transform.

• enable_verify (bool, optional) – Whether to verify metadata before transforming. Defaults to True.

Returns:

The transformed list of metadata dictionaries.

Return type:

List[Dict[str, Any]]

__repr__() → str

Transform string representation. Should be able to recreate class from this string.

Returns:

Transform representation.

Return type:

str

__str__() → str

Return str(self).

add_parent(parent) → None

Add parent Seedable object and set up RNGs accordingly

get_second_seed(seed: int) → int

Gets second seed, usually used to seed both torch and numpy generators with slightly different seeds

Parameters:

seed (int) – Seed to use.

Returns:

New seed.

Return type:

int

seed(seed: int) → None

Seed number generators with given seed.

Parameters:

seed (int) – Seed to use.

setup_rngs() → None

Initialize torch and numpy number generators, and update its children.

update(signal: Signal | DatasetSignal) → None

Update bookeeping for signals

Parameters:

signal (Signal | DatasetSignal) – signal to update metadata.

Raises:

NotImplementedError – Inherited classes must override this method.

update_from_parent() → None

Update numpy and torch number generators with parent seed