diff --git a/src/pyuwds3/types/scene_node.py b/src/pyuwds3/types/scene_node.py index 54a3562..ef99bb4 100644 --- a/src/pyuwds3/types/scene_node.py +++ b/src/pyuwds3/types/scene_node.py @@ -50,7 +50,8 @@ def __init__(self, p_cov_h=0.85, m_cov_h=1e-9, p_cov_p=0.8, m_cov_p=0.01, p_cov_r=0.06, m_cov_r=0.001, - time=None): + time=None, + agent=False): """ Scene node constructor """ self.id = str(uuid.uuid4()).replace("-", "") @@ -60,6 +61,8 @@ def __init__(self, self.bbox = None self.mask = None + self.agent=agent + self.n_init = n_init self.max_lost = max_lost self.max_age = max_age