diff --git a/models/common.py b/models/common.py index 4e03b9c7e47e..75cc4e97bbc7 100644 --- a/models/common.py +++ b/models/common.py @@ -23,6 +23,18 @@ import torch.nn as nn from PIL import Image from torch.cuda import amp + +# Import 'ultralytics' package or install if if missing +try: + import ultralytics + + assert hasattr(ultralytics, '__version__') # verify package is not directory +except (ImportError, AssertionError): + import os + + os.system('pip install -U ultralytics') + import ultralytics + from ultralytics.utils.plotting import Annotator, colors, save_one_box from utils import TryExcept diff --git a/utils/general.py b/utils/general.py index 8a265c741b91..017103752ef7 100644 --- a/utils/general.py +++ b/utils/general.py @@ -35,6 +35,16 @@ import torch import torchvision import yaml + +# Import 'ultralytics' package or install if if missing +try: + import ultralytics + + assert hasattr(ultralytics, '__version__') # verify package is not directory +except (ImportError, AssertionError): + os.system('pip install -U ultralytics') + import ultralytics + from ultralytics.utils.checks import check_requirements from utils import TryExcept, emojis