From f8ad84ccb4694d2112ec534b72b85bee8050f7e8 Mon Sep 17 00:00:00 2001 From: Hadrien Mary Date: Sat, 3 Nov 2018 15:11:41 -0400 Subject: [PATCH] Return results and predictions after inference --- maskrcnn_benchmark/engine/inference.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/maskrcnn_benchmark/engine/inference.py b/maskrcnn_benchmark/engine/inference.py index 752526982..da776d94a 100644 --- a/maskrcnn_benchmark/engine/inference.py +++ b/maskrcnn_benchmark/engine/inference.py @@ -426,3 +426,6 @@ def inference( check_expected_results(results, expected_results, expected_results_sigma_tol) if output_folder: torch.save(results, os.path.join(output_folder, "coco_results.pth")) + + return results, coco_results, predictions +