florent/geo-fastai
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Show function with quantile per band

Browse Source
This commit is contained in:
Florent Guiotte 2022-04-29 15:59:56 +00:00
parent 54d0a10132
commit e4aeabc0fa
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
geofastai/raster.py
View File

@ -10,13 +10,18 @@ class RasterBase(fai.TensorImage):
def create(cls, fname, **kwargs):
data = rio.open(fname).read()
return cls(data)
def band_quantile(self, q):
return self.reshape(self.shape[0], -1, 1).quantile(q, axis=1, keepdims=True)
class RasterVNIR(RasterBase):
def show(self, ctx=None, bands=[3,2,1], **kwargs):
def show(self, ctx=None, bands=[3,2,1], q=0.001, **kwargs):
im = self[bands]
im = im.to(fai.torch.float)
im = im / im.quantile(.9)
vmin = im.band_quantile(q)
vmax = im.band_quantile(1 - q)
im = (im - vmin) / (vmax - vmin)
im = im.clip(0, 1)
return fai.show_image(im, ctx=ctx, vmax=300, **kwargs)