Update export.py, yolo.py sys.path.append() (#3579)

This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-06-10 15:35:22 +02:00
1 parent 095197bd4a
commit 53ed872c28
2 files changed
+11 -7

No files matched your search

+4 -2
View File
@@ -10,8 +10,8 @@ import sys
from copy import deepcopy
from pathlib import Path
sys.path.append(Path(__file__).parent.parent.absolute().__str__()) # to run '$ python *.py' files in subdirectories
logger = logging.getLogger(__name__)
FILE = Path(__file__).absolute()
sys.path.append(FILE.parents[1].as_posix()) # add yolov5/ to path
from models.common import *
from models.experimental import *
@@ -25,6 +25,8 @@ try:
except ImportError:
thop = None
logger = logging.getLogger(__name__)
class Detect(nn.Module):
stride = None # strides computed during build