110 lines
3.0 KiB
Python
110 lines
3.0 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
HyTools: Hyperspectral image processing library
|
|
Copyright (C) 2021 University of Wisconsin
|
|
|
|
Authors: Adam Chlus, Zhiwei Ye, Philip Townsend.
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, version 3 of the License.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
This module contain functions for generating boolean masks specific to apply image corrections and
|
|
models.
|
|
"""
|
|
|
|
from scipy.ndimage.morphology import binary_erosion
|
|
import numpy as np
|
|
from .cloud import zhai_cloud
|
|
|
|
|
|
def ndi(hy_obj,args):
|
|
mask = hy_obj.ndi(args['band_1'],args['band_2'])
|
|
mask = (mask >= float(args['min'])) & (mask <= float(args['max']))
|
|
return mask
|
|
|
|
|
|
def ancillary(hy_obj,args):
|
|
''' Mask ancillary datasets based off min and max threshold
|
|
|
|
'''
|
|
if args['name'] == 'cosine_i':
|
|
mask= hy_obj.cosine_i()
|
|
else:
|
|
mask = hy_obj.get_anc(args['name'])
|
|
mask = (mask >= float(args['min'])) & (mask <= float(args['max']))
|
|
return mask
|
|
|
|
|
|
def neon_edge(hy_obj,args):
|
|
'''
|
|
Mask artifacts in NEON images around edges.
|
|
'''
|
|
radius =args['radius']
|
|
y_grid, x_grid = np.ogrid[-radius: radius + 1, -radius: radius + 1]
|
|
window = (x_grid**2 + y_grid**2 <= radius**2).astype(np.float32)
|
|
buffer_edge = binary_erosion(hy_obj.mask['no_data'], window).astype(bool)
|
|
return buffer_edge
|
|
|
|
|
|
def kernel_finite(hy_obj,args):
|
|
'''
|
|
Create NDVI bin class mask
|
|
'''
|
|
k_vol = hy_obj.volume_kernel(hy_obj.brdf['volume'])
|
|
k_geom = hy_obj.geom_kernel(hy_obj.brdf['geometric'],
|
|
b_r=hy_obj.brdf["b/r"],
|
|
h_b =hy_obj.brdf["h/b"])
|
|
mask = np.isfinite(k_vol) & np.isfinite(k_geom)
|
|
return mask
|
|
|
|
|
|
def cloud(hy_obj,args):
|
|
if args['method'] == 'zhai_2018':
|
|
mask = ~zhai_cloud(hy_obj,args['cloud'],args['shadow'],
|
|
args['T1'], args['t2'], args['t3'],
|
|
args['t4'], args['T7'], args['T8'])
|
|
|
|
return mask
|
|
|
|
|
|
def water(hy_obj,args):
|
|
'''
|
|
Create water mask using NDWI threshold
|
|
'''
|
|
mask = hy_obj.ndi(args['band_1'],args['band_2'])
|
|
mask = mask <= float(args['threshold'])
|
|
|
|
mask = binary_erosion(mask)
|
|
|
|
return mask
|
|
|
|
|
|
def external(hy_obj,args):
|
|
'''Load a mask from an external dataset
|
|
'''
|
|
|
|
hy_obj.anc_path['external_mask'] = [args['files'][hy_obj.file_name], 0]
|
|
mask = hy_obj.get_anc('external_mask') == args['class']
|
|
|
|
return mask
|
|
|
|
|
|
def band(hy_obj,args):
|
|
'''
|
|
Create mask using band thresholds
|
|
'''
|
|
|
|
mask = hy_obj.get_wave(args['band'])
|
|
mask = (mask >= float(args['min'])) & (mask <= float(args['max']))
|
|
|
|
return mask
|