|
10 | 10 | from ..data_transfer_objects.eval_function import EvalFunctionEntry |
11 | 11 | from ..errors import EvalFunctionNotAvailableError |
12 | 12 |
|
| 13 | +# TODO(gunnar) split up into modules |
| 14 | +# pylint: disable=too-many-lines |
| 15 | + |
13 | 16 |
|
14 | 17 | class PolygonIOUConfig(EvalFunctionConfig): |
15 | 18 | def __call__( |
@@ -66,6 +69,8 @@ def __call__( |
66 | 69 | enforce_label_match=enforce_label_match, |
67 | 70 | iou_threshold=iou_threshold, |
68 | 71 | confidence_threshold=confidence_threshold, |
| 72 | + annotation_filters=annotation_filters, |
| 73 | + prediction_filters=prediction_filters, |
69 | 74 | **kwargs, |
70 | 75 | ) |
71 | 76 |
|
@@ -123,6 +128,8 @@ def __call__( |
123 | 128 | """ |
124 | 129 | return super().__call__( |
125 | 130 | iou_threshold=iou_threshold, |
| 131 | + annotation_filters=annotation_filters, |
| 132 | + prediction_filters=prediction_filters, |
126 | 133 | **kwargs, |
127 | 134 | ) |
128 | 135 |
|
@@ -186,6 +193,8 @@ def __call__( |
186 | 193 | enforce_label_match=enforce_label_match, |
187 | 194 | iou_threshold=iou_threshold, |
188 | 195 | confidence_threshold=confidence_threshold, |
| 196 | + annotation_filters=annotation_filters, |
| 197 | + prediction_filters=prediction_filters, |
189 | 198 | **kwargs, |
190 | 199 | ) |
191 | 200 |
|
@@ -249,6 +258,8 @@ def __call__( |
249 | 258 | enforce_label_match=enforce_label_match, |
250 | 259 | iou_threshold=iou_threshold, |
251 | 260 | confidence_threshold=confidence_threshold, |
| 261 | + annotation_filters=annotation_filters, |
| 262 | + prediction_filters=prediction_filters, |
252 | 263 | **kwargs, |
253 | 264 | ) |
254 | 265 |
|
@@ -312,6 +323,8 @@ def __call__( |
312 | 323 | enforce_label_match=enforce_label_match, |
313 | 324 | iou_threshold=iou_threshold, |
314 | 325 | confidence_threshold=confidence_threshold, |
| 326 | + annotation_filters=annotation_filters, |
| 327 | + prediction_filters=prediction_filters, |
315 | 328 | **kwargs, |
316 | 329 | ) |
317 | 330 |
|
@@ -369,6 +382,8 @@ def __call__( |
369 | 382 | """ |
370 | 383 | return super().__call__( |
371 | 384 | iou_threshold=iou_threshold, |
| 385 | + annotation_filters=annotation_filters, |
| 386 | + prediction_filters=prediction_filters, |
372 | 387 | **kwargs, |
373 | 388 | ) |
374 | 389 |
|
@@ -432,6 +447,8 @@ def __call__( |
432 | 447 | enforce_label_match=enforce_label_match, |
433 | 448 | iou_threshold=iou_threshold, |
434 | 449 | confidence_threshold=confidence_threshold, |
| 450 | + annotation_filters=annotation_filters, |
| 451 | + prediction_filters=prediction_filters, |
435 | 452 | **kwargs, |
436 | 453 | ) |
437 | 454 |
|
@@ -495,6 +512,8 @@ def __call__( |
495 | 512 | enforce_label_match=enforce_label_match, |
496 | 513 | iou_threshold=iou_threshold, |
497 | 514 | confidence_threshold=confidence_threshold, |
| 515 | + annotation_filters=annotation_filters, |
| 516 | + prediction_filters=prediction_filters, |
498 | 517 | **kwargs, |
499 | 518 | ) |
500 | 519 |
|
@@ -794,7 +813,10 @@ def __call__( |
794 | 813 | Finally, the most outer list combines these filters as a disjunction (OR). |
795 | 814 | """ |
796 | 815 | return super().__call__( |
797 | | - confidence_threshold=confidence_threshold, f1_method=f1_method |
| 816 | + confidence_threshold=confidence_threshold, |
| 817 | + f1_method=f1_method, |
| 818 | + annotation_filters=annotation_filters, |
| 819 | + prediction_filters=prediction_filters, |
798 | 820 | ) |
799 | 821 |
|
800 | 822 | @classmethod |
|
0 commit comments