Upload 3 files

README.md

@@ -1,13 +1,9 @@
----
-title: Objectdetection
-emoji: 🏆
-colorFrom: purple
-colorTo: pink
-sdk: gradio
-sdk_version: 5.24.0
-app_file: app.py
-pinned: false
-short_description: objectdetection
----
+# GeoAI Mask R-CNN Object Detector
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+This app uses a custom Mask R-CNN model trained with the GeoAI library to detect objects in aerial/satellite imagery. 
+Upload a 512x512 GeoTIFF tile to see the detected objects as a GeoJSON file.
+
+## Instructions
+1. Upload a GeoTIFF tile.
+2. Click submit.
+3. Download the resulting GeoJSON with detected objects.

app.py

@@ -0,0 +1,33 @@
+import gradio as gr
+import geoai
+import torch
+
+model_path = "best_model.pth"
+output_mask = "prediction.tif"
+output_vector = "prediction.geojson"
+
+def detect_objects(input_image):
+    input_image.save("input.tif")
+
+    geoai.object_detection(
+        image_path="input.tif",
+        output_path=output_mask,
+        model_path=model_path,
+        window_size=512,
+        overlap=256,
+        confidence_threshold=0.5,
+        batch_size=4,
+        num_channels=3,
+    )
+
+    geoai.orthogonalize(output_mask, output_vector, epsilon=2)
+
+    return output_vector
+
+gr.Interface(
+    fn=detect_objects,
+    inputs=gr.Image(type="pil", label="Upload GeoTIFF"),
+    outputs="file",
+    title="Object Detection with GeoAI Mask R-CNN",
+    description="Upload a 512x512 GeoTIFF to detect objects using a custom-trained Mask R-CNN model."
+).launch()

requirements.txt

@@ -0,0 +1,3 @@
+gradio
+geoai-py
+torch