Unified '/project/name' results saving (#1377)
* Project/name update * Update ci-testing.yml * address project with path separator failure mode * Project/name update * address project with path separator failure mode * Update ci-testing.yml * detect.py default --name bug fix * missing rstrip PR * train/exp0 to train/exp
This commit is contained in:
1 parent
04081f8102
commit
c4addd7761
7 files changed
+71
-78
No files matched your search
+11
-17
@@ -60,7 +60,7 @@ def init_seeds(seed=0):
|
||||
init_torch_seeds(seed)
|
||||
|
||||
|
||||
def get_latest_run(search_dir='./runs'):
|
||||
def get_latest_run(search_dir='.'):
|
||||
# Return path to most recent 'last.pt' in /runs (i.e. to --resume from)
|
||||
last_list = glob.glob(f'{search_dir}/**/last*.pt', recursive=True)
|
||||
return max(last_list, key=os.path.getctime) if last_list else ''
|
||||
@@ -951,23 +951,17 @@ def output_to_target(output, width, height):
|
||||
return np.array(targets)
|
||||
|
||||
|
||||
def increment_dir(dir, comment=''):
|
||||
# Increments a directory runs/exp1 --> runs/exp2_comment
|
||||
n = 0 # number
|
||||
dir = str(Path(dir)) # os-agnostic
|
||||
if os.path.isdir(dir):
|
||||
stem = ''
|
||||
dir += os.sep # removed by Path
|
||||
def increment_path(path, exist_ok=True, sep=''):
|
||||
# Increment path, i.e. runs/exp --> runs/exp{sep}0, runs/exp{sep}1 etc.
|
||||
path = Path(path) # os-agnostic
|
||||
if (path.exists() and exist_ok) or (not path.exists()):
|
||||
return str(path)
|
||||
else:
|
||||
stem = Path(dir).stem
|
||||
|
||||
dirs = sorted(glob.glob(dir + '*')) # directories
|
||||
if dirs:
|
||||
matches = [re.search(r"%s(\d+)" % stem, d) for d in dirs]
|
||||
idxs = [int(m.groups()[0]) for m in matches if m]
|
||||
if idxs:
|
||||
n = max(idxs) + 1 # increment
|
||||
return dir + str(n) + ('_' + comment if comment else '')
|
||||
dirs = glob.glob(f"{path}{sep}*") # similar paths
|
||||
matches = [re.search(rf"%s{sep}(\d+)" % path.stem, d) for d in dirs]
|
||||
i = [int(m.groups()[0]) for m in matches if m] # indices
|
||||
n = max(i) + 1 if i else 2 # increment number
|
||||
return f"{path}{sep}{n}" # update path
|
||||
|
||||
|
||||
# Plotting functions ---------------------------------------------------------------------------------------------------
|
||||
|
||||
Reference in new issue
Block a user