(self)
| 28 | self.cat_bank = np.zeros((self.nID), dtype=np.int) |
| 29 | |
| 30 | def reset(self): |
| 31 | self.id_count = 0 |
| 32 | self.nID = 10000 |
| 33 | self.tracks = [] |
| 34 | self.embedding_bank = np.zeros((self.nID, 128)) |
| 35 | self.cat_bank = np.zeros((self.nID), dtype=np.int) |
| 36 | self.tracklet_ages = np.zeros((self.nID), dtype=np.int) |
| 37 | self.alive = [] |
| 38 | |
| 39 | def step(self, results_with_low, public_det=None): |
| 40 | results = [item for item in results_with_low if item['score'] >= self.opt.track_thresh] |