(gts, ts)
| 20 | |
| 21 | |
| 22 | def compare_dataframes(gts, ts): |
| 23 | accs = [] |
| 24 | names = [] |
| 25 | for k, tsacc in ts.items(): |
| 26 | if k in gts: |
| 27 | logger.info('Comparing {}...'.format(k)) |
| 28 | accs.append(mm.utils.compare_to_groundtruth(gts[k], tsacc, 'iou', distth=0.5)) |
| 29 | names.append(k) |
| 30 | else: |
| 31 | logger.warning('No ground truth for {}, skipping.'.format(k)) |
| 32 | |
| 33 | return accs, names |
| 34 | |
| 35 | |
| 36 | # evaluate MOTA |