Spaces:
Running
Running
| from tracker import update_tracker | |
| import cv2 | |
| class baseDet(object): | |
| def __init__(self): | |
| self.img_size = 640 | |
| self.threshold = 0.3 | |
| self.stride = 1 | |
| def build_config(self): | |
| self.faceTracker = {} | |
| self.faceClasses = {} | |
| self.faceLocation1 = {} | |
| self.faceLocation2 = {} | |
| self.frameCounter = 0 | |
| self.currentCarID = 0 | |
| self.recorded = [] | |
| self.font = cv2.FONT_HERSHEY_SIMPLEX | |
| def feedCap(self, im): | |
| retDict = { | |
| 'frame': None, | |
| 'faces': None, | |
| 'list_of_ids': None, | |
| 'face_bboxes': [] | |
| } | |
| self.frameCounter += 1 | |
| im, faces, face_bboxes = update_tracker(self, im, self.frameCounter) | |
| retDict['frame'] = im | |
| retDict['faces'] = faces | |
| retDict['face_bboxes'] = face_bboxes | |
| return retDict | |
| def init_model(self): | |
| raise EOFError("Undefined model type.") | |
| def preprocess(self): | |
| raise EOFError("Undefined model type.") | |
| def detect(self): | |
| raise EOFError("Undefined model type.") | |