Update app.py

app.py

@@ -158,31 +158,19 @@ def get_transform_from_tif(tif_path):
         crs = src.crs
     return transform, crs
     
-def mask_to_polygons(mask, transform, target_class_id):
-    """
-    Convert a single-class mask to polygons with proper geospatial reference.
-    
-    Args:
-        mask (np.ndarray): Predicted mask (H, W) with class indices.
-        transform (Affine): Georeferencing transform from the input TIF.
-        target_class_id (int): The class ID to convert into polygons.
-        
-    Returns:
-        gpd.GeoDataFrame: Geodataframe containing polygons for the target class.
-    """
-    # Binary mask for the target class
-    binary_mask = (mask == target_class_id).astype(np.uint8)
+def mask_to_polygons(mask, transform, mining_class_id=1):
+    # Create a binary mask for mining areas
+    mining_mask = (mask == mining_class_id).astype(np.uint8)
 
-    # Extract polygon shapes
+    # Extract shapes (polygons) from the mask
     results = (
-        {'properties': {'class': f'class_{target_class_id}'}, 'geometry': geom}
-        for geom, value in shapes(binary_mask, mask=None, transform=transform)
-        if value == 1
+        {'properties': {'class': 'mining_land'}, 'geometry': s}
+        for s, v in shapes(mining_mask, mask=None, transform=transform)
+        if v == 1  # Only keep areas marked as mining
     )
 
+    # Create a GeoDataFrame from the polygon results
     geoms = list(results)
-
-    # Return empty GeoDataFrame if no shapes found
     if not geoms:
         return gpd.GeoDataFrame(columns=['geometry'], geometry='geometry', crs='EPSG:4326')
 
@@ -191,21 +179,23 @@ def mask_to_polygons(mask, transform, target_class_id):
     return gdf
 
 
-def calculate_area(mining_gdf, wiup_gdf):
-    # Ensure both are in same CRS
+# Function to calculate illegal mining area (mining outside WIUP)
+def calculate_illegal_area(mining_gdf, wiup_gdf):
+    # Ensure both are in the same CRS
     mining_gdf = mining_gdf.to_crs(wiup_gdf.crs)
 
-    # Same = parts of mining polygons 
-    area_gdf = gpd.overlay(mining_gdf, wiup_gdf, how='intersection')
+    # Perform overlay operation: difference (mining area outside WIUP)
+    illegal_area_gdf = gpd.overlay(mining_gdf, wiup_gdf, how='difference')
 
     # Convert to metric CRS (e.g., UTM) to calculate area in m²
-    metric_crs = 'EPSG:4326'  # UTM zone for SE Asia (adjust based on location)
-    area_gdf = area_gdf.to_crs(metric_crs)
-    
-    area_gdf['area_m2'] = area_gdf.geometry.area
-    total_area = area_gdf['area_m2'].sum()
+    metric_crs = ' EPSG:4326'  # UTM zone for SE Asia (adjust based on location)
+    illegal_area_gdf = illegal_area_gdf.to_crs(metric_crs)
+
+    # Calculate the area in square meters
+    illegal_area_gdf['area_m2'] = illegal_area_gdf.geometry.area
+    total_illegal_area = illegal_area_gdf['area_m2'].sum()
 
-    return total_area, area_gdf
+    return total_illegal_area, illegal_area_gdf
 
 # Streamlit App Title
 st.title("Satellite Mining Segmentation: SAR + Optic Image Inference")
@@ -365,28 +355,19 @@ if st.button("Run Inference"):
             transform, crs = get_transform_from_tif(optic_path)
             
             # Convert mask to mining polygons
-            non_mining_gdf = mask_to_polygons(pred_label_mask, transform, target_class_id=0)
-            mining_gdf = mask_to_polygons(pred_label_mask, transform, target_class_id=1)
-            beach_gdf = mask_to_polygons(pred_label_mask, transform, target_class_id=2)
-
-            st.success(f"Non-mining polygons: {len(non_mining_gdf)}")
+            mining_gdf = mask_to_polygons(pred_label_mask, transform, mining_class_id=1)
             st.success(f"Mining polygons: {len(mining_gdf)}")
-            st.success(f"Beach polygons: {len(beach_gdf)}")
+
             
             # Make sure WIUP and prediction are in same CRS
             wiup_gdf = wiup_gdf.to_crs(mining_gdf.crs)
             st.success(f"WIUP CRS: {wiup_gdf.crs}")
             
             # Find area
-            non_mining_area, non_mining_area_gdf = calculate_area(non_mining_gdf, wiup_gdf)
-            mining_area, mining_area_gdf = calculate_area(mining_gdf, wiup_gdf)
-            beach_area, beach_area_gdf = calculate_area(beach_gdf, wiup_gdf)
+            mining_area, mining_area_gdf = calculate_illegal_area(mining_gdf, wiup_gdf)
             
             # Display in Streamlit
-            st.success(f" Non Mining Area : {non_mining_area/1e6:.2f} sq. km")
-            st.success(f" Mining Area : {mining_area/1e6:.2f} sq. km")
-            st.success(f" Beach Area : {beach_area/1e6:.2f} sq. km")
-                       
+            st.success(f" Illegal Mining Area : {mining_area/1e6:.2f} sq. km")                  
             
         else:
             st.warning("Please upload all three .tiff files to proceed.")