mirrors/yolov5
1
0
Fork 0
mirror of https://github.com/ultralytics/yolov5.git synced 2025-06-03 14:49:29 +08:00
Code Issues Projects Releases Wiki Activity

Update increment_path() to handle file paths (#2867)

Browse Source
This commit is contained in:
Glenn Jocher 2021-04-20 19:47:07 +02:00 committed by GitHub
parent 1df8c6c963
commit c5c647e281
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
utils/general.py
View File

@ -591,14 +591,16 @@ def apply_classifier(x, model, img, im0):
return x return x
def increment_path(path, exist_ok=True, sep=''): def increment_path(path, exist_ok=False, sep=''):
# Increment path, i.e. runs/exp --> runs/exp{sep}0, runs/exp{sep}1 etc. # Increment file or directory path, i.e. runs/exp --> runs/exp{sep}2, runs/exp{sep}3, ... etc.
path = Path(path) # os-agnostic path = Path(path) # os-agnostic
if (path.exists() and exist_ok) or (not path.exists()): if not path.exists() or exist_ok:
return str(path) return str(path)
else: else:
suffix = path.suffix
path = path.with_suffix('')
dirs = glob.glob(f"{path}{sep}*") # similar paths dirs = glob.glob(f"{path}{sep}*") # similar paths
matches = [re.search(rf"%s{sep}(\d+)" % path.stem, d) for d in dirs] 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 i = [int(m.groups()[0]) for m in matches if m] # indices
n = max(i) + 1 if i else 2 # increment number n = max(i) + 1 if i else 2 # increment number
return f"{path}{sep}{n}" # update path return f"{path}{sep}{n}{suffix}" # update path