DmitryRyumin markitantov commited on
Commit
d315e47
Β·
verified Β·
1 Parent(s): eac013f

Reset the tracked ID (#1)

Browse files

- Reset the tracked ID (28ee102466c043176ae81f9b87575b1cee4f628f)


Co-authored-by: Maxim Markitantov <[email protected]>

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: