Sfoglia il codice sorgente

[Fix] Removes the channel of monochrome image

Yizhou Chen 3 anni fa
parent
commit
83b3e1a843
1 ha cambiato i file con 1 aggiunte e 0 eliminazioni
  1. 1 0
      tools/utils/raster.py

+ 1 - 0
tools/utils/raster.py

@@ -158,6 +158,7 @@ class Raster:
             # the type is complex means this is a SAR data
             # the type is complex means this is a SAR data
             if isinstance(type(ima[0, 0]), complex):
             if isinstance(type(ima[0, 0]), complex):
                 ima = abs(ima)
                 ima = abs(ima)
+            ima = ima.squeeze()
         else:
         else:
             ima = ima.transpose((1, 2, 0))
             ima = ima.transpose((1, 2, 0))
         if self.to_uint8 is True:
         if self.to_uint8 is True: