Update relative `ROOT` logic (#4955)
* Update relative `ROOT` logic * python 3.9 Path().is_relative_to() removalpull/4956/head
parent
4e04cb0dc8
commit
5ed28603cf
|
@ -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
|
||||
|
|
|
@ -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)
|
||||
|
||||
|
|
Loading…
Reference in New Issue