fix doc & typing

utils.py

@@ -23,22 +23,25 @@ def reformat_for_plotting(
     boxes: npt.NDArray[np.float64],
     labels: npt.NDArray[np.int_],
     scores: npt.NDArray[np.float64],
-    shape: Tuple[int, int, int],
+    shape: Tuple[int, int],
     num_classes: int,
 ) -> Tuple[List[npt.NDArray[np.int_]], List[npt.NDArray[np.float64]]]:
     """
     Reformat YOLOX predictions for plotting.
+    - Unnormalizes boxes to original image size.
+    - Reformats boxes to [xmin, ymin, width, height].
+    - Converts to list of boxes and scores per class.
 
     Args:
-        boxes (np.ndarray): Array of bounding boxes.
-        labels (np.ndarray): Array of labels.
-        scores (np.ndarray): Array of confidence scores.
-        shape (tuple): Shape of the image.
+        boxes (np.ndarray [N, 4]): Array of bounding boxes in format [xmin, ymin, xmax, ymax].
+        labels (np.ndarray [N]): Array of labels.
+        scores (np.ndarray [N]): Array of confidence scores.
+        shape (tuple [2]): Shape of the image (height, width).
         num_classes (int): Number of classes.
 
     Returns:
-        list[np.ndarray]: List of box bounding boxes per class.
-        list[np.ndarray]: List of confidence scores per class.
+        list[np.ndarray[N]]: List of box bounding boxes per class.
+        list[np.ndarray[N]]: List of confidence scores per class.
     """
     boxes_plot = boxes.copy()
     boxes_plot[:, [0, 2]] *= shape[1]
@@ -56,6 +59,7 @@ def plot_sample(
     boxes_list: List[npt.NDArray[np.int_]],
     confs_list: List[npt.NDArray[np.float64]],
     labels: List[str],
+    show_text: bool = True,
 ) -> None:
     """
     Plots an image with bounding boxes.
@@ -66,6 +70,7 @@ def plot_sample(
         boxes_list (list[np.ndarray]): List of box bounding boxes per class.
         confs_list (list[np.ndarray]): List of confidence scores per class.
         labels (list): List of class labels.
+        show_text (bool, optional): Whether to show the text. Defaults to True.
     """
     plt.imshow(img, cmap="gray")
     plt.axis(False)
@@ -90,15 +95,16 @@ def plot_sample(
             plt.gca().add_patch(rect)
 
             # Add class and index label with proper alignment
-            plt.text(
-                box[0], box[1],
-                f"{l}_{box_idx}   conf={confs[box_idx]:.3f}",
-                color='white',
-                fontsize=8,
-                bbox=dict(facecolor=col, alpha=1, edgecolor=col, pad=0, linewidth=2),
-                verticalalignment='bottom',
-                horizontalalignment='left'
-            )
+            if show_text:
+                plt.text(
+                    box[0], box[1],
+                    f"{l}_{box_idx}   conf={confs[box_idx]:.3f}",
+                    color='white',
+                    fontsize=8,
+                    bbox=dict(facecolor=col, alpha=1, edgecolor=col, pad=0, linewidth=2),
+                    verticalalignment='bottom',
+                    horizontalalignment='left'
+                )
 
 
 def postprocess_preds_page_element(