diff --git a/openptv_python/parameters.py b/openptv_python/parameters.py index e8d0ecd..9929c8c 100644 --- a/openptv_python/parameters.py +++ b/openptv_python/parameters.py @@ -8,6 +8,7 @@ from openptv_python.constants import TR_MAX_CAMS +@dataclass class Parameters: """Base class for all parameters with a couple of methods.""" diff --git a/openptv_python/track.py b/openptv_python/track.py index a0dab39..db90c6f 100644 --- a/openptv_python/track.py +++ b/openptv_python/track.py @@ -189,7 +189,7 @@ def predict(prev_pos, curr_pos, output): output[1] = 2 * curr_pos[1] - prev_pos[1] -def pos3d_in_bounds(pos, bounds): +def pos3d_in_bounds(pos: np.ndarray, bounds: TrackPar) -> bool: """Check that all components of a pos3d are in their respective bounds. taken from a track_par object.