From 2af059c0d85f89813254a644443fb074033e3629 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 2 Apr 2021 11:55:10 +0200 Subject: [PATCH] PyTorch Hub model.save() increment as runs/hub/exp (#2684) * PyTorch Hub model.save() increment as runs/hub/exp This chane will align PyTorch Hub results saving with the existing unified results saving directory structure of runs/ /train /detect /test /hub /exp /exp2 ... * cleanup --- models/common.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/models/common.py b/models/common.py index a25172dcfcac..9970fbc8e2d9 100644 --- a/models/common.py +++ b/models/common.py @@ -11,7 +11,7 @@ from torch.cuda import amp from utils.datasets import letterbox -from utils.general import non_max_suppression, make_divisible, scale_coords, xyxy2xywh +from utils.general import non_max_suppression, make_divisible, scale_coords, increment_path, xyxy2xywh from utils.plots import color_list, plot_one_box from utils.torch_utils import time_synchronized @@ -324,9 +324,9 @@ def display(self, pprint=False, show=False, save=False, render=False, save_dir=' if show: img.show(self.files[i]) # show if save: - f = Path(save_dir) / self.files[i] - img.save(f) # save - print(f"{'Saving' * (i == 0)} {f},", end='' if i < self.n - 1 else ' done.\n') + f = self.files[i] + img.save(Path(save_dir) / f) # save + print(f"{'Saved' * (i == 0)} {f}", end=',' if i < self.n - 1 else f' to {save_dir}\n') if render: self.imgs[i] = np.asarray(img) @@ -337,8 +337,9 @@ def print(self): def show(self): self.display(show=True) # show results - def save(self, save_dir='results/'): - Path(save_dir).mkdir(exist_ok=True) + def save(self, save_dir='runs/hub/exp'): + save_dir = increment_path(save_dir, exist_ok=save_dir != 'runs/hub/exp') # increment save_dir + Path(save_dir).mkdir(parents=True, exist_ok=True) self.display(save=True, save_dir=save_dir) # save results def render(self):