(self, dets, meta)
| 204 | self.kalman_filter = KalmanFilter() |
| 205 | |
| 206 | def post_process(self, dets, meta): |
| 207 | dets = dets.detach().cpu().numpy() |
| 208 | dets = dets.reshape(1, -1, dets.shape[2]) |
| 209 | dets = ctdet_post_process( |
| 210 | dets.copy(), [meta['c']], [meta['s']], |
| 211 | meta['out_height'], meta['out_width'], self.opt.num_classes) |
| 212 | for j in range(1, self.opt.num_classes + 1): |
| 213 | dets[0][j] = np.array(dets[0][j], dtype=np.float32).reshape(-1, 5) |
| 214 | return dets[0] |
| 215 | |
| 216 | def merge_outputs(self, detections): |
| 217 | results = {} |
no outgoing calls
no test coverage detected