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:
Khiem DoanandGlenn Jocher authored and GitHub committed 2020-11-10 13:01:19 +01:00
1 parent 19e2482458
commit 1c8464e199
4 files changed
+13 -13

No files matched your search

+2 -2
View File
@@ -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