From a67cd184e21f91fe24bc313bf194f21cb07469b7 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 27 Apr 2021 16:48:53 +0200 Subject: [PATCH 1/2] Add yolov5/ to sys.path() for *.py subdir exec --- models/yolo.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/models/yolo.py b/models/yolo.py index d573c5a290e2..cbff70fc83d4 100644 --- a/models/yolo.py +++ b/models/yolo.py @@ -4,8 +4,9 @@ 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 @@ def parse_model(d, ch): # model_dict, input_channels(3) # 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) From 95047ae7e4dae79a3f84ab586e0584e51bca1ff3 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 27 Apr 2021 16:49:24 +0200 Subject: [PATCH 2/2] Update export.py --- models/export.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/models/export.py b/models/export.py index 38fefa7e891c..da15079149a1 100644 --- a/models/export.py +++ b/models/export.py @@ -7,8 +7,9 @@ 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