Skip to content

Commit 7314363

Browse files
authored
Add results.save(save_dir='path', exist_ok=False) (#9617)
Signed-off-by: Glenn Jocher <[email protected]> Signed-off-by: Glenn Jocher <[email protected]>
1 parent 1460e57 commit 7314363

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

models/common.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -775,12 +775,12 @@ def _run(self, pprint=False, show=False, save=False, crop=False, render=False, l
775775
def show(self, labels=True):
776776
self._run(show=True, labels=labels) # show results
777777

778-
def save(self, labels=True, save_dir='runs/detect/exp'):
779-
save_dir = increment_path(save_dir, exist_ok=save_dir != 'runs/detect/exp', mkdir=True) # increment save_dir
778+
def save(self, labels=True, save_dir='runs/detect/exp', exist_ok=False):
779+
save_dir = increment_path(save_dir, exist_ok, mkdir=True) # increment save_dir
780780
self._run(save=True, labels=labels, save_dir=save_dir) # save results
781781

782-
def crop(self, save=True, save_dir='runs/detect/exp'):
783-
save_dir = increment_path(save_dir, exist_ok=save_dir != 'runs/detect/exp', mkdir=True) if save else None
782+
def crop(self, save=True, save_dir='runs/detect/exp', exist_ok=False):
783+
save_dir = increment_path(save_dir, exist_ok, mkdir=True) if save else None
784784
return self._run(crop=True, save=save, save_dir=save_dir) # crop results
785785

786786
def render(self, labels=True):

0 commit comments

Comments
 (0)