Merge pull request #1986 from jprokos26/collect-outputs

`collect-outputs` util
twosixlabs · Nov 15, 2023 · c52d5bb · c52d5bb
2 parents 7d0eeec + 23a11d3
commit c52d5bb
Show file tree

Hide file tree

Showing 5 changed files with 897 additions and 0 deletions.
diff --git a/armory/cli/__init__.py b/armory/cli/__init__.py
@@ -1,3 +1,4 @@
+from armory.cli.tools.collect_outputs import collect_armory_outputs
 from armory.cli.tools.generate_shapes import generate_shapes
 from armory.cli.tools.log_current_branch import log_current_branch
 from armory.cli.tools.plot_mAP_by_giou_with_patch_cli import (
@@ -13,4 +14,8 @@
         plot_mAP_by_giou_with_patch_cli,
         "Visualize the output of the metric 'object_detection_AP_per_class_by_giou_from_patch.'",
     ),
+    "collect-outputs": (
+        collect_armory_outputs,
+        "Convert runs from the output directory into tables",
+    ),
 }