Update app.py

app.py

@@ -255,7 +255,7 @@ def get_detection_data(image_data):
         image = img
         print(f"Debug - Original image shape: {image.shape}")
         
-        image_segmentation, objects_data = img_seg.predict(image)
+        image_segmentation, objects_data = img_seg.predict(resized_image)
         depthmap, depth_colormap = depth_estimator.make_prediction(resized_image)
 
         detections = []
@@ -274,6 +274,7 @@ def get_detection_data(image_data):
                 print(f"Warning: Invalid depth value ({depth_value}) for Object {idx}. Using default depth of 1.0...")
                 depth_value = 1.0  # Default to 1.0 if depth is invalid or NaN
 
+            
             # Scale from 1536x1024 to 1512x1008, ensuring values fit within bounds
             scale_x = 1512 / 1536  # Width scaling factor
             scale_y = 1008 / 1024  # Height scaling factor
@@ -305,27 +306,27 @@ def get_detection_data(image_data):
             #"color": color_rgb,
             #"confidence": float(confidence)
 
-            #detections.append({
-                #"class_id": cls_id,
-                #"class_name": cls_name,
-                #"bounding_box": {
-                #    "vertices": get_box_vertices([x1, y1, x2, y2])
-                #},
-                #"center_2d": center,  # Add 2D center in screen space
-                #"distance": depth_value,  # Depth in meters
-                #"color": color_rgb,
-                #"confidence": float(confidence)
-                
             detections.append({
                 "class_id": cls_id,
                 "class_name": cls_name,
                 "bounding_box": {
-                    "vertices": scaled_vertices
+                    "vertices": get_box_vertices([x1, y1, x2, y2])
                 },
-                "center_2d": scaled_center,  # Use scaled center
-                "distance": float(depth_value),  # Ensure depth is a float, not NaN
+                "center_2d": center,  # Add 2D center in screen space
+                "distance": depth_value,  # Depth in meters
                 "color": color_rgb,
                 "confidence": float(confidence)
+                
+            #detections.append({
+                #"class_id": cls_id,
+                #"class_name": cls_name,
+                #"bounding_box": {
+                #    "vertices": scaled_vertices
+                #},
+                #"center_2d": scaled_center,  # Use scaled center
+                #"distance": float(depth_value),  # Ensure depth is a float, not NaN
+                #"color": color_rgb,
+                #"confidence": float(confidence)
             
     })