Skip to content

Commit c96e9e7

Browse files
authored
Merge pull request #2408 from JosuaRieder/no_console_results
Implement --no-console-results in inference.py
2 parents 63b2de7 + 4779063 commit c96e9e7

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

inference.py

+5-2
Original file line numberDiff line numberDiff line change
@@ -149,6 +149,8 @@
149149
help='include the class index in results')
150150
parser.add_argument('--exclude-output', action='store_true', default=False,
151151
help='exclude logits/probs from results, just indices. topk must be set !=0.')
152+
parser.add_argument('--no-console-results', action='store_true', default=False,
153+
help='disable printing the inference results to the console')
152154

153155

154156
def main():
@@ -348,8 +350,9 @@ def main():
348350
for fmt in args.results_format:
349351
save_results(df, results_filename, fmt)
350352

351-
print(f'--result')
352-
print(df.set_index(args.filename_col).to_json(orient='index', indent=4))
353+
if not args.no_console_results:
354+
print(f'--result')
355+
print(df.set_index(args.filename_col).to_json(orient='index', indent=4))
353356

354357

355358
def save_results(df, results_filename, results_format='csv', filename_col='filename'):

0 commit comments

Comments
 (0)