Update relative ROOT logic (#4955)

* Update relative `ROOT` logic

* python 3.9 Path().is_relative_to() removal
This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-09-27 18:26:41 -07:00
1 parent 4e04cb0dc8
commit 5ed28603cf
2 files changed
+4 -1

No files matched your search

+1
View File
@@ -20,6 +20,7 @@ FILE = Path(__file__).resolve()
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
import numpy as np
import tensorflow as tf
+3 -1
View File
@@ -15,11 +15,12 @@ FILE = Path(__file__).resolve()
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
from models.common import *
from models.experimental import *
from utils.autoanchor import check_anchor_order
from utils.general import check_yaml, make_divisible, set_logging
from utils.general import check_yaml, make_divisible, print_args, set_logging
from utils.plots import feature_visualization
from utils.torch_utils import copy_attr, fuse_conv_and_bn, initialize_weights, model_info, scale_img, \
select_device, time_sync
@@ -281,6 +282,7 @@ if __name__ == '__main__':
parser.add_argument('--profile', action='store_true', help='profile model speed')
opt = parser.parse_args()
opt.cfg = check_yaml(opt.cfg) # check YAML
print_args(FILE.stem, opt)
set_logging()
device = select_device(opt.device)