(gts, ts)
| 111 | |
| 112 | |
| 113 | def compare_dataframes(gts, ts): |
| 114 | accs = [] |
| 115 | names = [] |
| 116 | for k, tsacc in ts.items(): |
| 117 | if k in gts: |
| 118 | logger.info('Comparing {}...'.format(k)) |
| 119 | accs.append(mm.utils.compare_to_groundtruth(gts[k], tsacc, 'iou', distth=0.5)) |
| 120 | names.append(k) |
| 121 | else: |
| 122 | logger.warning('No ground truth for {}, skipping.'.format(k)) |
| 123 | |
| 124 | return accs, names |
| 125 | |
| 126 | |
| 127 | @logger.catch |