Add crs and transform for array input (#400)

* Add crs and transform for array input

* Update show_anns

Author: giswqs

samgeo/common.py

@@ -1458,12 +1458,16 @@ def array_to_image(
         array = cv2.imread(array)
         array = cv2.cvtColor(array, cv2.COLOR_BGR2RGB)
 
-    if output.endswith(".tif") and source is not None:
-        with rasterio.open(source) as src:
-            crs = src.crs
-            transform = src.transform
-            if compress is None:
-                compress = src.compression
+    if output.endswith(".tif"):
+        if source is not None:
+            with rasterio.open(source) as src:
+                crs = src.crs
+                transform = src.transform
+                if compress is None:
+                    compress = src.compression
+        else:
+            crs = kwargs.get("crs", None)
+            transform = kwargs.get("transform", None)
 
         # Determine the minimum and maximum values in the array
 

samgeo/samgeo.py

@@ -425,11 +425,11 @@ def show_anns(
             img[m] = color_mask
         ax.imshow(img)
 
-        if "dpi" not in kwargs:
-            kwargs["dpi"] = 100
+        # if "dpi" not in kwargs:
+        #     kwargs["dpi"] = 100
 
-        if "bbox_inches" not in kwargs:
-            kwargs["bbox_inches"] = "tight"
+        # if "bbox_inches" not in kwargs:
+        #     kwargs["bbox_inches"] = "tight"
 
         plt.axis(axis)
 
@@ -442,7 +442,7 @@ def show_anns(
                 )
             else:
                 array = self.annotations
-            array_to_image(array, output, self.source)
+            array_to_image(array, output, self.source, **kwargs)
 
     def set_image(self, image, image_format="RGB"):
         """Set the input image as a numpy array.

samgeo/samgeo2.py

@@ -456,11 +456,11 @@ def show_anns(
             img[m] = color_mask
         ax.imshow(img)
 
-        if "dpi" not in kwargs:
-            kwargs["dpi"] = 100
+        # if "dpi" not in kwargs:
+        #     kwargs["dpi"] = 100
 
-        if "bbox_inches" not in kwargs:
-            kwargs["bbox_inches"] = "tight"
+        # if "bbox_inches" not in kwargs:
+        #     kwargs["bbox_inches"] = "tight"
 
         plt.axis(axis)
 
@@ -473,7 +473,7 @@ def show_anns(
                 )
             else:
                 array = self.annotations
-            common.array_to_image(array, output, self.source)
+            common.array_to_image(array, output, self.source, **kwargs)
 
     @torch.no_grad()
     def set_image(