Use pathlib instead of low-level module (#1329)
* Use pathlib instead of low-level module * Use pathlib instead of low-level module * Update detect.py * Update test.py * reformat Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
1 parent
19e2482458
commit
1c8464e199
4 files changed
+13
-13
No files matched your search
@@ -47,9 +47,9 @@ def test(data,
|
||||
|
||||
# Directories
|
||||
if save_dir == Path('runs/test'): # if default
|
||||
os.makedirs('runs/test', exist_ok=True) # make base
|
||||
save_dir.mkdir(parents=True, exist_ok=True) # make base
|
||||
save_dir = Path(increment_dir(save_dir / 'exp', opt.name)) # increment run
|
||||
os.makedirs(save_dir / 'labels' if save_txt else save_dir, exist_ok=True) # make new dir
|
||||
(save_dir / 'labels' if save_txt else save_dir).mkdir(parents=True, exist_ok=True) # make new dir
|
||||
|
||||
# Load model
|
||||
model = attempt_load(weights, map_location=device) # load FP32 model
|
||||
|
||||
Reference in new issue
Block a user