(self)
| 159 | self.trackers.append(tracker) |
| 160 | |
| 161 | def run(self): |
| 162 | while True: |
| 163 | if not self.paused: |
| 164 | ret, self.frame = self.cap.read() |
| 165 | if not ret: |
| 166 | break |
| 167 | frame_gray = cv2.cvtColor(self.frame, cv2.COLOR_BGR2GRAY) |
| 168 | for tracker in self.trackers: |
| 169 | tracker.update(frame_gray) |
| 170 | |
| 171 | vis = self.frame.copy() |
| 172 | for tracker in self.trackers: |
| 173 | tracker.draw_state(vis) |
| 174 | if len(self.trackers) > 0: |
| 175 | cv2.imshow('tracker state', self.trackers[-1].state_vis) |
| 176 | self.rect_sel.draw(vis) |
| 177 | |
| 178 | cv2.imshow('frame', vis) |
| 179 | ch = cv2.waitKey(10) |
| 180 | if ch == 27: |
| 181 | break |
| 182 | if ch == ord(' '): |
| 183 | self.paused = not self.paused |
| 184 | if ch == ord('c'): |
| 185 | self.trackers = [] |
| 186 | |
| 187 | |
| 188 | if __name__ == '__main__': |
no test coverage detected