Update periodontitis_detection.py

periodontitis_detection.py

@@ -216,11 +216,7 @@ class SimpleDentalSegmentationNoEnhance:
         Perform full analysis on a dental image:
         segmentation, detection, distance measurement, and visualization.
         """
-        img_bgr = cv2.imread(image_path)
-        if img_bgr is None:
-            raise FileNotFoundError(f"Could not read image: {image_path}")
 
-        h_orig, w_orig = img_bgr.shape[:2]
         cej_unet, abc_unet, _ = self.run_unet(img_bgr)
         cej_mask = self.resize_mask_to_original(cej_unet, (h_orig, w_orig))
         abc_mask = self.resize_mask_to_original(abc_unet, (h_orig, w_orig))
@@ -235,6 +231,20 @@ class SimpleDentalSegmentationNoEnhance:
         alpha = 0.4
         combined = cv2.addWeighted(overlay, alpha, img_bgr, 1 - alpha, 0)
 
+        for det in detections:
+            x1, y1, x2, y2 = det["bbox"].astype(int)
+            cv2.rectangle(combined, (x1, y1), (x2, y2), (0, 255, 0), 2)
+            cv2.putText(
+                combined,
+                f"Tooth {det['tooth_id']}",
+                (x1, y1 - 5),
+                cv2.FONT_HERSHEY_SIMPLEX,
+                0.5,
+                (0, 255, 0),
+                1,
+                cv2.LINE_AA
+            )
+    
         all_results = []
 
         for det in detections: