Skip to content

Commit 6d30b82

Browse files
zhima771rwightman
authored andcommitted
fix: create results_dir if missing before saving results
1 parent 954613a commit 6d30b82

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

inference.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -345,6 +345,7 @@ def main():
345345
results_filename = f'{args.model}-{img_size}'
346346

347347
if args.results_dir:
348+
os.makedirs(args.results_dir, exist_ok=True)
348349
results_filename = os.path.join(args.results_dir, results_filename)
349350

350351
for fmt in args.results_format:

0 commit comments

Comments
 (0)