Commit 2d31ced3 authored by Narcon Nicolas's avatar Narcon Nicolas
Browse files

Use RCS algo for rasterio example

parent b2fe0f84
No related merge requests found
Showing with 20 additions and 6 deletions
+20 -6
import numpy as np from scipy import ndimage
from skimage.transform import rescale
import skimage.color as color
def rcs_pansharpen(xs, pan):
"""
Implemented by me
cf https://www.sfpt.fr/hyperspectral/wp-content/uploads/2018/01/May_panorama_pansharpening.pdf
"""
xs_zoom = ndimage.zoom(xs, zoom=4)
pan_low_pass = ndimage.gaussian_filter(pan, sigma=1)
pansharpened = xs_zoom * (pan / pan_low_pass)
return pansharpened
## NOT USED ANYMORE (can not compare speed with OTB)
"""
# functions # functions
def stretch(bands, lower_percent=2, higher_percent=98): def stretch(bands, lower_percent=2, higher_percent=98):
out = np.zeros_like(bands) out = np.zeros_like(bands)
...@@ -96,4 +109,5 @@ def pansharpen(m, pan, method='browley', W=0.1, all_data=False): ...@@ -96,4 +109,5 @@ def pansharpen(m, pan, method='browley', W=0.1, all_data=False):
if all_data: if all_data:
return rgbn_scaled, image, I return rgbn_scaled, image, I
else: else:
return image return image
\ No newline at end of file """
\ No newline at end of file
import rasterio import rasterio
from functions_for_rasterio import pansharpen from functions_for_rasterio import rcs_pansharpen
xs_path = 'xs.tif' xs_path = 'xs.tif'
pan_path = 'pan.vrt' pan_path = 'pan.vrt'
...@@ -17,5 +17,5 @@ print(xs.shape) ...@@ -17,5 +17,5 @@ print(xs.shape)
print(pan.shape) print(pan.shape)
# Pansharpening # Pansharpening
pxs = pansharpen(xs, pan) pxs = rcs_pansharpen(xs, pan)
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment