commit

yolov6/utils/general.py

@@ -0,0 +1,24 @@
+#!/usr/bin/env python3
+# -*- coding:utf-8 -*-
+import os
+import glob
+from pathlib import Path
+
+def increment_name(path):
+    "increase save directory's id"
+    path = Path(path)
+    sep = ''
+    if path.exists():
+        path, suffix = (path.with_suffix(''), path.suffix) if path.is_file() else (path, '')
+        for n in range(1, 9999):
+            p = f'{path}{sep}{n}{suffix}'
+            if not os.path.exists(p):
+                break
+        path = Path(p)
+    return path
+
+
+def find_latest_checkpoint(search_dir='.'):
+    # Find the most recent saved checkpoint in search_dir
+    checkpoint_list = glob.glob(f'{search_dir}/**/last*.pt', recursive=True)
+    return max(checkpoint_list, key=os.path.getctime) if checkpoint_list else ''