diplomat.wx_gui.score_lib.ScoreEngine

class diplomat.wx_gui.score_lib.ScoreEngine

Bases: ABC

Represents a “ScoreEngine” which assigns scores to every frame that can be displayed in the UI.

__init__()

Methods

__init__()
compute_bad_indexes(scores)	Given all the scores, return a numpy array of integers specifying the indexes of frames which score "bad" or poorly for this metric.
compute_scores(poses, prog_bar[, sub_section])	Compute the scores for the given set of poses.
get_name()
get_settings()	Return a single setting, being the config for this single widget...

abstract compute_bad_indexes(scores: ndarray) → ndarray

Given all the scores, return a numpy array of integers specifying the indexes of frames which score “bad” or poorly for this metric. Typically done via a configurable threshold.

abstract compute_scores(poses: Pose, prog_bar: ProgressBar, sub_section: slice | None = None) → ndarray

Compute the scores for the given set of poses.

Parameters:

• poses – The predicted locations/probabilities for given body parts.

• prog_bar – A progress bar for keeping track of current progress on computing the scores.

• sub_section – Optional slice, passed when only some scores need to be recomputed.

Returns:

A numpy array of floats between 1 and 0, being the length of the video…

abstract get_settings() → SettingCollection

Return a single setting, being the config for this single widget…