|
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