Update calculations.py

calculations.py

@@ -1,6 +1,5 @@
 import math
 import numpy as np
-import matplotlib.pyplot as plt
 import cv2
 from tensorflow.keras import preprocessing
 
@@ -11,88 +10,68 @@ def convert_to_real_measurements(pixel_measurement, pixel_height, real_height_cm
     height_ratio = real_height_cm / pixel_height
     return pixel_measurement * height_ratio
 
-def measure_body_sizes(side_colored_mask, front_colored_mask, sideposes, frontposes, real_height_cm, rainbow):
+def measure_body_sizes(side_colored_mask, front_colored_mask, sideposes, frontposes, real_height_cm, rainbow=None):
     """Measure various body sizes based on detected poses."""
     measurements = []
     
+    # Check if poses are available
+    if not frontposes or len(frontposes) == 0:
+        return measurements
+
     for pose in frontposes:
-        # Assuming each `pose` is a dictionary with 'keypoints' that are already in the required format
-        keypoints = pose[0]  # This should directly give us the dictionary
+        keypoints = pose[0]  # Directly accessing keypoints
 
-        # Extract positions directly from keypoints
-        left_eye = keypoints[1].position
-        right_eye = keypoints[2].position
-        nose = keypoints[3].position
-        right_ear = keypoints[4].position
-        left_shoulder = keypoints[5].position
-        right_shoulder = keypoints[6].position
-        left_elbow = keypoints[7].position
-        right_elbow = keypoints[8].position
-        left_wrist = keypoints[9].position
-        right_wrist = keypoints[10].position
-        left_hip = keypoints[11].position
-        right_hip = keypoints[12].position
-        left_knee = keypoints[13].position
-        right_knee = keypoints[14].position
-        left_ankle = keypoints[15].position
-        right_ankle = keypoints[16].position
-        
-        # Calculate pixel height (from the top of the head to the bottom of the ankle)
-        pixel_height = euclidean_distance((left_eye.x, left_eye.y), (left_ankle.x, left_ankle.y))
-        
+        # Extracting position coordinates
+        try:
+            left_eye = (keypoints[1].position.x, keypoints[1].position.y)
+            right_eye = (keypoints[2].position.x, keypoints[2].position.y)
+            left_shoulder = (keypoints[5].position.x, keypoints[5].position.y)
+            right_shoulder = (keypoints[6].position.x, keypoints[6].position.y)
+            left_wrist = (keypoints[9].position.x, keypoints[9].position.y)
+            left_hip = (keypoints[11].position.x, keypoints[11].position.y)
+            right_hip = (keypoints[12].position.x, keypoints[12].position.y)
+            left_ankle = (keypoints[15].position.x, keypoints[15].position.y)
+            right_ankle = (keypoints[16].position.x, keypoints[16].position.y)
+        except (IndexError, AttributeError):
+            continue
+
+        # Calculate pixel height
+        pixel_height = euclidean_distance(left_eye, left_ankle)
         
+        # Calculate measurements
         shoulder_width_cm = convert_to_real_measurements(
-            euclidean_distance((left_shoulder.x, left_shoulder.y),(right_shoulder.x, right_shoulder.y)),
+            euclidean_distance(left_shoulder, right_shoulder),
             pixel_height, real_height_cm
         )
 
-        # arm_length_cm = convert_to_real_measurements(
-        #     euclidean_distance((right_shoulder.x, right_shoulder.y), (right_elbow.x, right_elbow.y)),
-        #     pixel_height, real_height_cm
-        # ) + convert_to_real_measurements(
-        #     euclidean_distance((right_elbow.x, right_elbow.y), (right_wrist.x, right_wrist.y)),
-        #     pixel_height, real_height_cm
-        # )
-        
-        # leg_length_cm = convert_to_real_measurements(
-        #     euclidean_distance((left_hip.x, left_hip.y), (left_knee.x, left_knee.y)),
-        #     pixel_height, real_height_cm
-        # ) + convert_to_real_measurements(
-        #     euclidean_distance((left_knee.x, left_knee.y), (left_ankle.x, left_ankle.y)),
-        #     pixel_height, real_height_cm
-        # )
-        
         arm_length_cm = convert_to_real_measurements(
-            euclidean_distance((left_shoulder.x, left_shoulder.y), (left_wrist.x, left_wrist.y)),
+            euclidean_distance(left_shoulder, left_wrist),
             pixel_height, real_height_cm
-        ) 
+        )
         
         leg_length_cm = convert_to_real_measurements(
-            euclidean_distance((left_hip.x, left_hip.y), (left_ankle.x, right_ankle.y)),
+            euclidean_distance(left_hip, left_ankle),
             pixel_height, real_height_cm
         )
 
         shoulder_to_waist_cm = convert_to_real_measurements(
-            euclidean_distance((left_shoulder.x, left_shoulder.y), (left_hip.x, left_hip.y)),
+            euclidean_distance(left_shoulder, left_hip),
             pixel_height, real_height_cm
         )
 
-        # Calculate waist circumference using the ellipse circumference formula
-        a = euclidean_distance((left_hip.x, left_hip.y), (right_hip.x, right_hip.y)) / 2
-        # b = euclidean_distance((), ()) / 2
+        # Simplified waist circumference - needs more accurate implementation
+        waist_width = euclidean_distance(left_hip, right_hip)
+        waist_circumference_cm = convert_to_real_measurements(
+            waist_width, pixel_height, real_height_cm
+        ) * 2  # Rough estimate
 
-        # Use Ramanujan's approximation for the circumference of an ellipse
-        # waist_circumference_px = math.pi * (3*(a + b) - math.sqrt((3*a + b)*(a + 3*b)))
-        waist_circumference_cm = 90 #convert_to_real_measurements(waist_circumference_px, pixel_height, real_height_cm)
-
-        
-        # Convert pixel measurements to real measurements using the height ratio
         measurements.append({
             "height_cm": real_height_cm,
-            "arm_length_cm":  arm_length_cm,
+            "arm_length_cm": arm_length_cm,
             "shoulder_width_cm": shoulder_width_cm,
             "shoulder_to_waist_cm": shoulder_to_waist_cm,
             "leg_length_cm": leg_length_cm,
+            "waist_circumference_cm": waist_circumference_cm
         })
 
-    return measurements
+    return measurements