nicetoolbox.detectors.method_detectors.head_orientation.spiga_detector.Spiga

class nicetoolbox.detectors.method_detectors.head_orientation.spiga_detector.Spiga(config, io, data)[source]

Bases: BaseDetector

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 the SPIGA method detector with all inference preparation.

Parameters:
  • config (dict) – Configuration settings for SPIGA.

  • io (class) – IO class instance for input/output operations.

  • data (class) – Data class instance for frame and subject data.

Methods

post_inference

Calculate head orientation in 2D image after SPIGA inference.

run_inference

Runs the inference of the method detector in a separate terminal/cmd window using the specified virtual environment or conda environment.

visualization

Abstract method to visualize the output of the method, preferably as a video.

Attributes

algorithm

components

post_inference()[source]

Calculate head orientation in 2D image after SPIGA inference.

run_inference() None

Runs the inference of the method detector in a separate terminal/cmd window using the specified virtual environment or conda environment. Captures the output and logs the success or failure of the inference.

visualization(data)[source]

Abstract method to visualize the output of the method, preferably as a video.

This method is intended to generate a visual representation of the method detector’s output. The visualization should be saved in the self.viz_folder.

Parameters:

data (any) – The data to be visualized. The type and content of this parameter depend on the specific implementation of the method detector.

Returns:

None. This method does not return any value. However, it should save the

visualization in the self.viz_folder.

Raises:

NotImplementedError – If this method is not implemented in the derived classes.