Spaces:
Running
on
Zero
Running
on
Zero
Reset the tracked ID
Browse filesOn the local computer, after several runs, you have to reset the tracker.
- 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:
|