zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Fix relative `ROOT` Pytorch Hub custom model bug (#4974) 

* Fix relative `ROOT` Pytorch Hub custom model bug

* Update yolo.py
This commit is contained in:
Glenn Jocher 2021-09-28 06:36:12 -07:00 committed by GitHub
parent c1bed601e9
commit fb982d6030
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
models/tf.py
View File

@ -20,7 +20,7 @@ FILE = Path(__file__).resolve()
ROOT = FILE.parents[1] # YOLOv5 root directory ROOT = FILE.parents[1] # YOLOv5 root directory
if str(ROOT) not in sys.path: if str(ROOT) not in sys.path:
sys.path.append(str(ROOT)) # add ROOT to PATH sys.path.append(str(ROOT)) # add ROOT to PATH
ROOT = ROOT.relative_to(Path.cwd()) # relative # ROOT = ROOT.relative_to(Path.cwd()) # relative
import numpy as np import numpy as np
import tensorflow as tf import tensorflow as tf

2
models/yolo.py
View File

@ -15,7 +15,7 @@ FILE = Path(__file__).resolve()
ROOT = FILE.parents[1] # YOLOv5 root directory ROOT = FILE.parents[1] # YOLOv5 root directory
if str(ROOT) not in sys.path: if str(ROOT) not in sys.path:
sys.path.append(str(ROOT)) # add ROOT to PATH sys.path.append(str(ROOT)) # add ROOT to PATH
ROOT = ROOT.relative_to(Path.cwd()) # relative # ROOT = ROOT.relative_to(Path.cwd()) # relative
from models.common import * from models.common import *
from models.experimental import * from models.experimental import *