From 45654d0755a8d84b47e87c11b4dbff1763a0fa22 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 27 Apr 2021 17:02:07 +0200 Subject: [PATCH] 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 | 3 ++- models/yolo.py | 5 +++-- 2 files changed, 5 insertions(+), 3 deletions(-) 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 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)