Update export.py, yolo.py sys.path.append() (#3579)
This commit is contained in:
1 parent
095197bd4a
commit
53ed872c28
2 files changed
+11
-7
No files matched your search
+4
-2
@@ -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
|
||||
|
||||
Reference in new issue
Block a user