Reset the tracked ID

#1
by markitantov - opened
Files changed (1) hide show
  1. app/load_models.py +3 -0
app/load_models.py CHANGED
@@ -862,6 +862,9 @@ class VideoFeatureExtractor:
862
  def preprocess_video(self) -> None:
863
  counter = 0
864
 
 
 
 
865
  while True:
866
  ret, frame = self.cap.read()
867
  if not ret:
 
862
  def preprocess_video(self) -> None:
863
  counter = 0
864
 
865
+ if hasattr(self.model_loader.face_model.predictor, 'trackers'):
866
+ self.model_loader.face_model.predictor.trackers[0].reset()
867
+
868
  while True:
869
  ret, frame = self.cap.read()
870
  if not ret: