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

* Fix relative `ROOT` Pytorch Hub custom model bug * Update yolo.py
fhkiel-mlaip · Sep 28, 2021 · 0b48222 · 0b48222
1 parent 29c348a
commit 0b48222
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/models/tf.py b/models/tf.py
@@ -20,7 +20,7 @@
 ROOT = FILE.parents[1]  # YOLOv5 root directory
 if str(ROOT) not in sys.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 tensorflow as tf

diff --git a/models/yolo.py b/models/yolo.py
@@ -15,7 +15,7 @@
 ROOT = FILE.parents[1]  # YOLOv5 root directory
 if str(ROOT) not in sys.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.experimental import *