Skip to content

tensoriou

tensoriou

Module for TensorIou TensorMetric

Classes

TensorIoU 

TensorIoU(key, threshold=0.5)

Bases: TensorMetric

TensorMetric for calculating the IoU

Source code in niceml/mlcomponents/resultanalyzers/tensors/tensoriou.py 
20
21
22
23
24
25
26
27
28
29
def __init__(
    self,
    key: str,
    threshold: float = 0.5,
):
    super().__init__(key)
    self.threshold = threshold
    self.data_description = None
    self.intersection_sum = None
    self.union_sum = None