Update periodontitis_detection.py

periodontitis_detection.py

@@ -226,6 +226,8 @@ class SimpleDentalSegmentationNoEnhance:
         abc_mask = self.resize_mask_to_original(abc_unet, (h_orig, w_orig))
         detections = self.detect_teeth(img_bgr)
 
+        print(f"Detected {len(detections)} teeth")
+
         # Create a color overlay for CEJ (red) and ABC (blue)
         overlay = img_bgr.copy()
         overlay[cej_mask == 1] = (0, 0, 255)   # Red for CEJ
@@ -239,7 +241,7 @@ class SimpleDentalSegmentationNoEnhance:
 
         for det in detections:
             x1, y1, x2, y2 = det["bbox"].astype(int)
-            cv2.rectangle(combined, (x1, y1), (x2, y2), (0, 255, 0), 2)
+            cv2.rectangle(combined, (x1, y1), (x2, y2), (0, 255, 0), 3)
             cv2.putText(
                 combined,
                 f"Tooth {det['tooth_id']}",