sys append fixes

yolov5/detect.py

@@ -14,7 +14,8 @@
 import torch.backends.cudnn as cudnn
 
 FILE = Path(__file__).absolute()
-sys.path.append(FILE.parents[0].as_posix())  # add yolov5/ to path
+# appending pip package root directory to path to support legacy way of running script
+sys.path.append(FILE.parent.parent.as_posix())
 
 from yolov5.models.experimental import attempt_load
 from yolov5.utils.datasets import LoadStreams, LoadImages

yolov5/export.py

@@ -14,7 +14,8 @@
 from torch.utils.mobile_optimizer import optimize_for_mobile
 
 FILE = Path(__file__).absolute()
-sys.path.append(FILE.parents[0].as_posix())  # add yolov5/ to path
+# appending pip package root directory to path to support legacy way of running script
+sys.path.append(FILE.parent.parent.as_posix())
 
 from yolov5.models.common import Conv
 from yolov5.models.yolo import Detect

yolov5/test.py

@@ -17,7 +17,8 @@
 from tqdm import tqdm
 
 FILE = Path(__file__).absolute()
-sys.path.append(FILE.parents[0].as_posix())  # add yolov5/ to path
+# appending pip package root directory to path to support legacy way of running script
+sys.path.append(FILE.parent.parent.as_posix())
 
 from yolov5.models.experimental import attempt_load
 from yolov5.utils.datasets import create_dataloader

yolov5/train.py

@@ -30,7 +30,8 @@
 from tqdm import tqdm
 
 FILE = Path(__file__).absolute()
-sys.path.append(FILE.parent.parent.as_posix())  # add yolov5/ to path
+# appending pip package root directory to path to support legacy way of running script
+sys.path.append(FILE.parent.parent.as_posix())
 
 from yolov5 import test  # for end-of-epoch mAP
 from yolov5.models.experimental import attempt_load