nicetoolbox.detectors.method_detectors.spiga.spiga_detector

SPIGA method detector class.

Functions

extract_key_per_value

Extracts keys from a dictionary based on the type of their values.

return_direction_vector

Classes

Spiga

SPIGA is a method detector that computes the head_orientation component.

class nicetoolbox.detectors.method_detectors.spiga.spiga_detector.Spiga(io: SequenceIO, data: SequenceData, sequence_context: SequenceRuntimeConfig, algorithm_instance: str)[source]

SPIGA is a method detector that computes the head_orientation component.

Component: head_orientation

components

A list containing the name of the component: head_orientation

Type:

list

algorithm

Algorithm name used to compute the head_orientation component.

Type:

str

camera_names

List of camera names used to capture original input data.

Type:

list

Initialize base method detector with references.

post_inference()[source]

Calculate head orientation in 2D image after SPIGA inference.

visualization(data)[source]

Visualize detector output.

Parameters:

data – Output from run() or external data source

nicetoolbox.detectors.method_detectors.spiga.spiga_detector.extract_key_per_value(input_dict)[source]

Extracts keys from a dictionary based on the type of their values.

If all values in the dictionary are integers, it returns a list of keys. If any value is a list, it appends an index to the key to create a unique key.

Parameters:

input_dict (dict) – The input dictionary to extract keys from.

Returns:

A list of keys extracted from the input dictionary.

Return type:

return_keys (list)

Raises:
  • NotImplementedError – If a value in the dictionary is neither an integer nor a

  • list.