提交 42e78ab2 编写于 作者: M Max Frei 提交者: Waleed

Added automated epoch recognition for Windows. (#798)

Unified regex expression for both, Linux and Windows.
上级 bde3fb5c
...@@ -2259,8 +2259,11 @@ class MaskRCNN(): ...@@ -2259,8 +2259,11 @@ class MaskRCNN():
if model_path: if model_path:
# Continue from we left of. Get epoch and date from the file name # Continue from we left of. Get epoch and date from the file name
# A sample model path might look like: # A sample model path might look like:
# /path/to/logs/coco20171029T2315/mask_rcnn_coco_0001.h5 # \path\to\logs\coco20171029T2315\mask_rcnn_coco_0001.h5 (Windows)
regex = r".*/[\w-]+(\d{4})(\d{2})(\d{2})T(\d{2})(\d{2})/mask\_rcnn\_[\w-]+(\d{4})\.h5" # /path/to/logs/coco20171029T2315/mask_rcnn_coco_0001.h5 (Linux)
regex = r".*[/\\][\w-]+(\d{4})(\d{2})(\d{2})T(\d{2})(\d{2})[/\\]mask\_rcnn\_[\w-]+(\d{4})\.h5"
m = re.match(regex, model_path) m = re.match(regex, model_path)
if m: if m:
now = datetime.datetime(int(m.group(1)), int(m.group(2)), int(m.group(3)), now = datetime.datetime(int(m.group(1)), int(m.group(2)), int(m.group(3)),
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册