fix type annotations

This commit is contained in:
kirilllzaitsev committed 2023-07-19 22:54:39 +02:00
1 parent 2c0905ba28
commit 950744111a
1 file changed
+6 -4
+6 -4
View File
@@ -1,7 +1,9 @@
from __future__ import annotations
from dataclasses import dataclass
from typing import Callable, List, Optional, Tuple
import matplotlib
import matplotlib
import matplotlib.pyplot as plt
import numpy as np
@@ -32,7 +34,7 @@ class ConfusionMatrix:
conf_threshold: float,
iou_threshold: float,
classes: Optional[List[str]],
) -> "ConfusionMatrix":
) -> ConfusionMatrix:
"""
Create ConfusionMatrix from matrix.
@@ -60,7 +62,7 @@ class ConfusionMatrix:
classes: List[str],
conf_threshold: float = 0.3,
iou_threshold: float = 0.5,
) -> "ConfusionMatrix":
) -> ConfusionMatrix:
"""
Calculate confusion matrix based on predicted and ground-truth detections.
@@ -219,7 +221,7 @@ class ConfusionMatrix:
callback: Callable[[np.ndarray], sv.Detections],
conf_threshold: float = 0.3,
iou_threshold: float = 0.5,
) -> "ConfusionMatrix":
) -> ConfusionMatrix:
"""
Create confusion matrix from dataset and callback function.