add null crs case

coolzhao · Aug 13, 2023 · 06aaf06 · 06aaf06
1 parent 53b7cbe
commit 06aaf06
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/GeoSAM-Image-Encoder/geosam/image_encoder.py b/GeoSAM-Image-Encoder/geosam/image_encoder.py
@@ -186,16 +186,16 @@ def encode_image(
                 extent = [i for i in src.bounds]
                 extent_crs = src.crs
             else:
-                extent_crs = CRS.from_user_input(
-                    extent.split(' ')[-1].strip()[1:-1]
-                )
+                extent_crs = extent.split(' ')[-1].strip()[1:-1].strip()
                 extent = [float(i.strip(',').strip())
                           for i in extent.split(' ')[:-1]]
-                extent = warp.transform_bounds(
-                    extent_crs,
-                    src.crs,
-                    *extent,
-                )
+                if extent_crs != '':
+                    extent_crs = CRS.from_user_input(extent_crs)
+                    extent = warp.transform_bounds(
+                        extent_crs,
+                        src.crs,
+                        *extent,
+                    )
 
         # check bands
         if bands is None: