Reference for ultralytics/models/__init__.py#
Improvements
This page is sourced from https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/__init__.py. Have an improvement or example to add? Open a Pull Request — thank you! 🙏
Summary
Function ultralytics.models.__getattr__#
def __getattr__(name)Lazy-import SAM so standard YOLO imports don't load optional torchvision internals.
ultralytics/models/__init__.py
def __getattr__(name):
"""Lazy-import SAM so standard YOLO imports don't load optional torchvision internals."""
if name == "SAM":
# Scoped for import ultralytics speed: SAM pulls optional torchvision-heavy modules.
from .sam import SAM
return SAM
raise AttributeError(f"module {__name__} has no attribute {name}")