vyvuyubhji

main.py

@@ -51,9 +51,11 @@ async def predict_image(file: UploadFile = File(...)):
         print(len(results))
         
         background_element = next((element for element in results if element['label'] == 'background'), None)
+        print(background_element["mask"].size)
         
         if background_element:
             background_area = calculate_mask_area(background_element['mask'], True)
+            print(background_area)
         else:
             background_area = 0
         

utils/helpers.py

@@ -11,8 +11,5 @@ def image_to_base64(image: Image.Image) -> str:
 
 def calculate_mask_area(mask: Image.Image, background=False) -> int:
     mask_array = np.array(mask)
-    if background:
-        non_zero_pixels = np.count_nonzero(mask_array == 0)
-    else:
-        non_zero_pixels = np.count_nonzero(mask_array)
+    non_zero_pixels = np.count_nonzero(mask_array)
     return non_zero_pixels