| 298 | postprocess_time * 1000)) |
| 299 | |
| 300 | def report(self, average=False): |
| 301 | dic = {} |
| 302 | pre_time = self.preprocess_time_s.value() |
| 303 | infer_time = self.inference_time_s.value() |
| 304 | post_time = self.postprocess_time_s.value() |
| 305 | track_time = self.tracking_time_s.value() |
| 306 | |
| 307 | dic['preprocess_time_s'] = round(pre_time / max(1, self.img_num), |
| 308 | 4) if average else pre_time |
| 309 | dic['inference_time_s'] = round(infer_time / max(1, self.img_num), |
| 310 | 4) if average else infer_time |
| 311 | dic['postprocess_time_s'] = round(post_time / max(1, self.img_num), |
| 312 | 4) if average else post_time |
| 313 | dic['img_num'] = self.img_num |
| 314 | total_time = pre_time + infer_time + post_time |
| 315 | if self.with_tracker: |
| 316 | dic['tracking_time_s'] = round(track_time / max(1, self.img_num), |
| 317 | 4) if average else track_time |
| 318 | total_time = total_time + track_time |
| 319 | dic['total_time_s'] = round(total_time, 4) |
| 320 | return dic |
| 321 | |
| 322 | |
| 323 | def get_current_memory_mb(): |