Skip to content

Detection

Bases: TargetModel

Source code in autodistill/detection/detection_target_model.py
 8
 9
10
11
12
13
14
15
16
17
18
19
class DetectionTargetModel(TargetModel):
    @abstractmethod
    def __init__(self):
        pass

    @abstractmethod
    def predict(self, input: str, confidence: float = 0.5) -> sv.Detections:
        pass

    @abstractmethod
    def train(self):
        pass