models/detectors/yolov8_defence.py

from models.detectors.yolov8 import HuggingFaceYoloV8Detector


class HuggingFaceYoloV8DefenceDetector(HuggingFaceYoloV8Detector):
    """YOLOv8m detector fine-tuned on defence data hosted on Hugging Face."""

    REPO_ID = "spencercdz/YOLOv8m_defence"
    WEIGHT_FILE = "yolov8m_defence.pt"

    def __init__(self, score_threshold: float = 0.3) -> None:
        super().__init__(score_threshold=score_threshold)
        self.name = "hf_yolov8_defence"