add from_matrix()

This commit is contained in:
kirilllzaitsev 2023-07-07 22:29:17 +02:00
parent 476be527e5
commit 2b876bdd64
1 changed files with 26 additions and 0 deletions

View File

@ -17,6 +17,32 @@ class ConfusionMatrix:
conf_threshold: float
iou_threshold: float
@classmethod
def from_matrix(
cls,
matrix,
conf_threshold: float,
iou_threshold: float,
classes: Optional[List[str]],
) -> "ConfusionMatrix":
"""
Create ConfusionMatrix from matrix.
Args:
matrix: confusion matrix.
classes: all known classes.
conf_threshold: detection confidence threshold between 0 and 1. Detections with lower confidence will be excluded.
iou_threshold: detection iou threshold between 0 and 1. Detections with lower iou will be classified as FP.
"""
classes = classes if classes is not None else [str(x) for x in (range(len(matrix)))]
return cls(
matrix=matrix,
classes=classes,
num_classes=len(classes),
conf_threshold=conf_threshold,
iou_threshold=iou_threshold,
)
@classmethod
def from_detections(
cls,