Spaces:
Running
on
Zero
Running
on
Zero
Reset the tracked ID (#1)
Browse files- Reset the tracked ID (28ee102466c043176ae81f9b87575b1cee4f628f)
Co-authored-by: Maxim Markitantov <[email protected]>
- 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:
|