there was a bug in the rasterio src handler from weecology@e6e2716 (w…

…eecology#591)

deepforest/utilities.py

@@ -293,9 +293,10 @@ def shapefile_to_annotations(shapefile,
     with rasterio.open(rgb) as src:
         left, bottom, right, top = src.bounds
         resolution = src.res[0]
+        raster_crs = src.crs
 
     # Check matching the crs
-    if not gdf.crs == src.crs:
+    if not gdf.crs == raster_crs:
         raise ValueError("The shapefile crs {} does not match the image crs {}".format(
             gdf.crs, src.crs))
 

tests/test_utilities.py

@@ -72,7 +72,18 @@ def test_shapefile_to_annotations(tmpdir):
     image_path = get_data("OSBS_029.tif")
     shp = utilities.shapefile_to_annotations(shapefile="{}/annotations.shp".format(tmpdir), rgb=image_path, savedir=tmpdir, geometry_type="bbox")
     assert shp.shape[0] == 2
+
+def test_shapefile_to_annotations_incorrect_crs(tmpdir):
+    sample_geometry = [geometry.Point(404211.9 + 10,3285102 + 20),geometry.Point(404211.9 + 20,3285102 + 20)]
+    labels = ["Tree","Tree"]
+    df = pd.DataFrame({"geometry":sample_geometry,"label":labels})
+    gdf = gpd.GeoDataFrame(df, geometry="geometry", crs="EPSG:32618")
+    gdf["geometry"] = [geometry.box(left, bottom, right, top) for left, bottom, right, top in gdf.geometry.buffer(0.5).bounds.values]
 
+    gdf.to_file("{}/annotations.shp".format(tmpdir))
+    image_path = get_data("OSBS_029.tif")
+    with pytest.raises(ValueError):
+        shp = utilities.shapefile_to_annotations(shapefile="{}/annotations.shp".format(tmpdir), rgb=image_path, savedir=tmpdir, geometry_type="bbox")
 def test_boxes_to_shapefile_projected(m):
     img = get_data("OSBS_029.tif")
     r = rio.open(img)