.

sorghum_pipeline/output/manager.py

@@ -55,7 +55,8 @@ class OutputManager:
         try:
             mask = plant_data.get('mask')
             if isinstance(mask, np.ndarray):
-                cv2.imwrite(str(results_dir / 'mask.png'), mask)
+                titled = self._add_title_banner(mask, 'Mask')
+                cv2.imwrite(str(results_dir / 'mask.png'), titled)
         except Exception as e:
             logger.error(f"Failed to save mask: {e}")
 
@@ -67,7 +68,8 @@ class OutputManager:
                 overlay = self._create_overlay(base_image, mask)
                 # Convert BGR→RGB for correct viewing in standard image viewers
                 overlay_rgb = cv2.cvtColor(overlay, cv2.COLOR_BGR2RGB)
-                cv2.imwrite(str(results_dir / 'overlay.png'), overlay_rgb)
+                titled = self._add_title_banner(overlay_rgb, 'Segmentation Overlay')
+                cv2.imwrite(str(results_dir / 'overlay.png'), titled)
         except Exception as e:
             logger.error(f"Failed to save overlay: {e}")
 
@@ -80,7 +82,8 @@ class OutputManager:
                     base_image = self._normalize_to_uint8(base_image.astype(np.float64))
                 # Convert BGR→RGB for human viewing
                 comp_rgb = cv2.cvtColor(base_image, cv2.COLOR_BGR2RGB)
-                cv2.imwrite(str(results_dir / 'composite.png'), comp_rgb)
+                titled = self._add_title_banner(comp_rgb, 'Composite (Segmentation Input)')
+                cv2.imwrite(str(results_dir / 'composite.png'), titled)
         except Exception as e:
             logger.error(f"Failed to save composite: {e}")
 
@@ -105,6 +108,7 @@ class OutputManager:
                         ax.set_axis_off()
                         ax.set_facecolor('white')
                         im = ax.imshow(masked, cmap=cmap, vmin=vmin, vmax=vmax)
+                        ax.set_title(f"{name}", fontsize=12, fontweight='bold', pad=8)
                         # add colorbar
                         cbar = fig.colorbar(im, ax=ax, fraction=0.046, pad=0.04)
                         cbar.ax.tick_params(labelsize=8)
@@ -130,6 +134,7 @@ class OutputManager:
                     ax.set_axis_off()
                     ax.set_facecolor('white')
                     im = ax.imshow(img, cmap='gray', vmin=0, vmax=255)
+                    ax.set_title('Texture: LBP (Green Band)', fontsize=12, fontweight='bold', pad=8)
                     cbar = fig.colorbar(im, ax=ax, fraction=0.046, pad=0.04)
                     cbar.ax.tick_params(labelsize=8)
                     plt.tight_layout()
@@ -146,7 +151,8 @@ class OutputManager:
             images = morph.get('images', {})
             size_img = images.get('size_analysis')
             if isinstance(size_img, np.ndarray) and size_img.size > 0:
-                cv2.imwrite(str(results_dir / 'size.size_analysis.png'), size_img)
+                titled = self._add_title_banner(size_img, 'Morphology Size')
+                cv2.imwrite(str(results_dir / 'size.size_analysis.png'), titled)
         except Exception as e:
             logger.error(f"Failed to save size analysis: {e}")
     
@@ -175,4 +181,44 @@ class OutputManager:
             normalized = (arr - arr.min()) / (ptp + 1e-6) * 255
         else:
             normalized = np.zeros_like(arr)
-        return np.clip(normalized, 0, 255).astype(np.uint8)
+        return np.clip(normalized, 0, 255).astype(np.uint8)
+
+    def _add_title_banner(self, image: np.ndarray, title: str) -> np.ndarray:
+        """Add a top banner with centered title text to an image using OpenCV.
+        Supports grayscale or color images; returns a BGR image.
+        """
+        if image is None or image.size == 0:
+            return image
+
+        # Ensure 3-channel BGR for drawing
+        if image.ndim == 2:
+            base_bgr = cv2.cvtColor(image, cv2.COLOR_GRAY2BGR)
+        elif image.ndim == 3 and image.shape[2] == 3:
+            base_bgr = image.copy()
+        elif image.ndim == 3 and image.shape[2] == 4:
+            base_bgr = cv2.cvtColor(image, cv2.COLOR_BGRA2BGR)
+        else:
+            # Fallback: normalize to uint8 then convert to BGR
+            norm = self._normalize_to_uint8(image.astype(np.float64))
+            base_bgr = cv2.cvtColor(norm, cv2.COLOR_GRAY2BGR)
+
+        h, w = base_bgr.shape[:2]
+        banner_height = max(30, int(0.08 * h))
+        banner = np.full((banner_height, w, 3), (245, 245, 245), dtype=np.uint8)
+
+        # Compose banner + image
+        composed = np.vstack([banner, base_bgr])
+
+        # Put centered title text
+        font = cv2.FONT_HERSHEY_SIMPLEX
+        font_scale = max(0.5, min(1.0, w / 800.0))
+        thickness = 1
+        text = str(title)
+        (tw, th), baseline = cv2.getTextSize(text, font, font_scale, thickness)
+        x = max(5, (w - tw) // 2)
+        y = (banner_height + th) // 2
+        # Slight shadow for readability
+        cv2.putText(composed, text, (x+1, y+1), font, font_scale, (0, 0, 0), thickness+1, cv2.LINE_AA)
+        cv2.putText(composed, text, (x, y), font, font_scale, (0, 80, 0), thickness+1, cv2.LINE_AA)
+
+        return composed