мем

# Разрешаем конкретные Ultralytics классы для "safe load" PyTorch 2.6+
# Файл выполняется автоматически при старте интерпретатора.

try:
    from torch.serialization import add_safe_globals
    # Список классов, которые часто появляются в weight-файлах Ultralytics
    from ultralytics.nn.tasks import (
        DetectionModel,
        SegmentationModel,
        PoseModel,
        ClassificationModel,
    )
    add_safe_globals([DetectionModel, SegmentationModel, PoseModel, ClassificationModel])
except Exception:
    # Если ultralytics ещё не установлена или импорт падает — пропускаем,
    # чтобы не ломать стартовую последовательность.
    # Это делает патч устойчивым к порядку инициализации.
    pass