Add yolov5/ to sys.path() for *.py subdir exec (#2949)

* Add yolov5/ to sys.path() for *.py subdir exec

* Update export.py

models/export.py

@@ -7,8 +7,9 @@ Usage:
 import argparse
 import sys
 import time
+from pathlib import Path
 
-sys.path.append('./')  # to run '$ python *.py' files in subdirectories
+sys.path.append(Path(__file__).parent.parent.absolute().__str__())  # to run '$ python *.py' files in subdirectories
 
 import torch
 import torch.nn as nn

models/yolo.py

@@ -4,8 +4,9 @@ import argparse
 import logging
 import sys
 from copy import deepcopy
+from pathlib import Path
 
-sys.path.append('./')  # to run '$ python *.py' files in subdirectories
+sys.path.append(Path(__file__).parent.parent.absolute().__str__())  # to run '$ python *.py' files in subdirectories
 logger = logging.getLogger(__name__)
 
 from models.common import *
@@ -267,7 +268,7 @@ if __name__ == '__main__':
     # Create model
     model = Model(opt.cfg).to(device)
     model.train()
-                        
+
     # Profile
     # img = torch.rand(8 if torch.cuda.is_available() else 1, 3, 320, 320).to(device)
     # y = model(img, profile=True)