fix type annotations

This commit is contained in:
kirilllzaitsev 2023-07-19 22:54:39 +02:00
parent 2c0905ba28
commit 950744111a
1 changed files with 6 additions and 4 deletions

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.