Spaces:
Running
on
Zero
Running
on
Zero
Reset the tracked ID
#1
by
markitantov
- opened
- 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:
|