Update app.py

app.py

@@ -73,7 +73,8 @@ def main():
         thresh_pil = process_image(selected_image_pil)
         st.image(thresh_pil, caption="Thresholded Image", use_column_width=True)
     
-        processed_img_path = process_with_yolo(selected_image_pil)
+        # processed_img_path = process_with_yolo(selected_image_pil)
+        processed_img_path = process_with_yolo(thresh_pil)
     
         if processed_img_path:
             st.image(processed_img_path, caption="Processed with YOLO", use_column_width=True)
@@ -149,7 +150,24 @@ def process_image(selected_image):
 
 
 
+def get_detected_boxes(txt_path, img_width, img_height):
+    with open(txt_path, 'r') as f:
+        lines = f.readlines()
+        boxes = []
+        for line in lines:
+            parts = list(map(float, line.strip().split()))
+            # Assuming the format is: class x_center y_center width height confidence
+            x_center, y_center, width, height = parts[1:5]
+            
+            # Convert to pixel values
+            x_center *= img_width
+            y_center *= img_height
+            width *= img_width
+            height *= img_height
+
+            boxes.append([x_center, y_center, width, height])
 
+    return boxes