(gts, ts)
| 50 | |
| 51 | |
| 52 | def compare_dataframes(gts, ts): |
| 53 | accs = [] |
| 54 | names = [] |
| 55 | for k, tsacc in ts.items(): |
| 56 | if k in gts: |
| 57 | logging.info('Comparing {}...'.format(k)) |
| 58 | accs.append(mm.utils.compare_to_groundtruth(gts[k], tsacc, 'iou', distth=0.5)) |
| 59 | names.append(k) |
| 60 | else: |
| 61 | logging.warning('No ground truth for {}, skipping.'.format(k)) |
| 62 | |
| 63 | return accs, names |
| 64 | |
| 65 | |
| 66 | if __name__ == '__main__': |