diff --git a/utils/__init__.py b/utils/__init__.py index 649b288b3588..2af1466f1f1d 100644 --- a/utils/__init__.py +++ b/utils/__init__.py @@ -1,3 +1,4 @@ +import sys from pathlib import Path import torch @@ -5,6 +6,8 @@ FILE = Path(__file__).absolute() ROOT = FILE.parents[1] # yolov5/ dir +if str(ROOT) not in sys.path: + sys.path.append(str(ROOT)) # add ROOT to PATH # Check YOLOv5 Annotator font font = 'Arial.ttf'