From 937023b5f7b4e2bbc4399e38d2247c91ec73a11a Mon Sep 17 00:00:00 2001
From: "Sajjad P. Savoji" <sj.pakdaman@gmail.com>
Date: Tue, 14 Nov 2023 15:05:00 -0500
Subject: [PATCH] Update track.py

need to change track state to confirmed if n_init == 1 in the constructor. I am conditioning based on hits and n_init.
---
 deep_sort/track.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/deep_sort/track.py b/deep_sort/track.py
index f638e9b3..a1f9c16f 100644
--- a/deep_sort/track.py
+++ b/deep_sort/track.py
@@ -72,7 +72,6 @@ def __init__(self, mean, covariance, track_id, n_init, max_age,
         self.age = 1
         self.time_since_update = 0
 
-        self.state = TrackState.Tentative
         self.features = []
         if feature is not None:
             self.features.append(feature)
@@ -80,6 +79,8 @@ def __init__(self, mean, covariance, track_id, n_init, max_age,
         self._n_init = n_init
         self._max_age = max_age
 
+        self.state = TrackState.Tentative if self.hits < self._n_init else TrackState.Confirmed
+
     def to_tlwh(self):
         """Get current position in bounding box format `(top left x, top left y,
         width, height)`.