Skip to content

M
maskrcnn
Commit e7edc294 authored by Hadrien Mary's avatar Hadrien Mary Committed by Francisco Massa
Browse files
Options

Return results and predictions after inference (#105)

parent 0527726b
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
maskrcnn_benchmark/engine/inference.py
View file @ e7edc294
...@@ -426,3 +426,6 @@ def inference( ...@@ -426,3 +426,6 @@ def inference(
check_expected_results(results, expected_results, expected_results_sigma_tol) check_expected_results(results, expected_results, expected_results_sigma_tol)
if output_folder: if output_folder:
torch.save(results, os.path.join(output_folder, "coco_results.pth")) torch.save(results, os.path.join(output_folder, "coco_results.pth"))
return results, coco_results, predictions
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment