refactor: ♻️ add create_ascii_table utility and update COCO coco_evaluation (#1324)

Signed-off-by: Onuralp SEZER <thunderbirdtr@gmail.com>

Author: onuralpszr

pyproject.toml

@@ -13,7 +13,6 @@ dependencies = [
   "pillow>=8.2.0",
   "pyyaml",
   "fire",
-  "terminaltables",
   "requests",
   "click",
   "torch>=2.4.1",

sahi/scripts/coco_evaluation.py

@@ -9,7 +9,8 @@
 
 import fire
 import numpy as np
-from terminaltables import AsciiTable
+
+from sahi.utils.table import create_ascii_table
 
 
 def _cocoeval_summarize(
@@ -316,8 +317,7 @@ def evaluate_core(
                 results_2d = itertools.zip_longest(*[results_flatten[i::num_columns] for i in range(num_columns)])
                 table_data = [headers]
                 table_data += [result for result in results_2d]
-                table = AsciiTable(table_data)
-                print("\n" + table.table)
+                print("\n" + create_ascii_table(table_data))
 
             if metric_items is None:
                 metric_items = ["mAP", "mAP50", "mAP75", "mAP_s", "mAP_m", "mAP_l", "mAP50_s", "mAP50_m", "mAP50_l"]

sahi/utils/table.py

@@ -0,0 +1,41 @@
+from __future__ import annotations
+
+from typing import Any
+
+
+def create_ascii_table(data: list[list[Any]]) -> str:
+    """
+    Creates a clean, properly padded ASCII string grid from a list of lists.
+
+    Args:
+        data (List[List[Any]]): A list of lists representing headers and rows.
+
+    Returns:
+        str: The formatted ASCII table as a string.
+    """
+    if not data or not data[0]:
+        return ""
+
+    # Convert all elements to strings, handling None values as empty strings
+    str_data = [[str(item) if item is not None else "" for item in row] for row in data]
+
+    # Calculate column widths
+    num_columns = max(len(row) for row in str_data)
+    col_widths = [0] * num_columns
+    for row in str_data:
+        for i, cell in enumerate(row):
+            if i < num_columns:
+                col_widths[i] = max(col_widths[i], len(cell))
+
+    # Define border part
+    border = "+" + "+".join("-" * (w + 2) for w in col_widths) + "+"
+
+    lines = [border]
+    for i, row in enumerate(str_data):
+        padded_row = row + [""] * (num_columns - len(row))
+        content_row = "| " + " | ".join(cell.ljust(w) for cell, w in zip(padded_row, col_widths)) + " |"
+        lines.append(content_row)
+        if i == 0 or i == len(str_data) - 1:
+            lines.append(border)
+
+    return "\n".join(lines)

tests/test_table.py

@@ -0,0 +1,35 @@
+from __future__ import annotations
+
+import unittest
+
+from sahi.utils.table import create_ascii_table
+
+
+class TestTable(unittest.TestCase):
+    def test_create_ascii_table_basic(self):
+        data = [
+            ["Model", "Params(M)", "Dataset"],
+            ["ResNet50", 25.6, "ImageNet"],
+            ["YOLOv8n", 4.5, "COCO"],
+            ["Swin-Transformer", 88.2, "ADE20K"],
+        ]
+        table = create_ascii_table(data)
+
+        # Verify basic structure
+        self.assertIn("ResNet50", table)
+        self.assertIn("Dataset", table)
+        self.assertTrue(table.startswith("+"))
+        self.assertTrue(table.endswith("+"))
+
+        # Verify alignment (Swin-Transformer is the longest model name)
+        # Content row should contain the full model name padded correctly.
+        self.assertIn("| Swin-Transformer |", table)
+
+    def test_empty_data(self):
+        self.assertEqual(create_ascii_table([]), "")
+        self.assertEqual(create_ascii_table([[]]), "")
+
+    def test_none_values(self):
+        data = [["ID", "Value"], [1, None]]
+        table = create_ascii_table(data)
+        self.assertIn("| 1  |       |", table)