allow geopandas input to shapefile to annotations

weecology · Dec 27, 2023 · e823400 · e823400
1 parent aa00ee1
commit e823400
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/deepforest/utilities.py b/deepforest/utilities.py
@@ -284,7 +284,10 @@ def shapefile_to_annotations(shapefile,
         results: a pandas dataframe
     """
     # Read shapefile
-    gdf = gpd.read_file(shapefile)
+    if isinstance(shapefile, str):
+        gdf = gpd.read_file(shapefile)
+    else:
+        gdf = shapefile
 
     # Determine geometry type and report to user
     if gdf.geometry.type.unique().shape[0] > 1: