Skip to content
GitLab
Commit c6790093 authored by mlang's avatar mlang
Browse files

Merge branch 'dev'

parents a8c2542e c149ede1
master
No related merge requests found
Hide whitespace changes
Inline Side-by-side
Showing
with 415 additions and 108 deletions
+415 -108
.gitignore
+ 6
− 0
View file @ c6790093
......@@ -60,3 +60,9 @@ data/l8_20130707\.tif\.aux\.xml
data/l8_20130831\.tif\.aux\.xml
qgis-auth\.db
ymraster/\.spyderproject
\.spyderworkspace
\.spyderproject
ymraster/driver_ext.py
+ 2
− 1
View file @ c6790093
......@@ -18,7 +18,8 @@ class DriverExt(GenericDriverExt):
drivername_map = {'.tif': 'GTiff',
'.h5': 'HDF5',
'.ers': 'ERS'}
'.ers': 'ERS',
'.vrt': 'VRT'}
__slots__ = ()
......
ymraster/raster_dtype.py
+ 100
− 42
View file @ c6790093
......@@ -2,13 +2,22 @@
try:
import otbApplication as otb
OTB_IS_LOAD = True
except ImportError as e:
raise ImportError(
str(e)
+ "\n\nPlease install Orfeo Toolbox if it isn't installed yet.\n\n"
OTB_IS_LOAD = False
ERROR_OTB = ImportError(str(e)
+ "\n\nThis functionnality is not available without Orfeo Toolbox. " +
"Please install Orfeo Toolbox if it isn't installed yet.\n\n"
"Also, add the otbApplication module path "
"(usually something like '/usr/lib/otb/python') "
"to the environment variable PYTHONPATH.\n")
"to the PYTHONPATH environment variable.")
# raise ImportError(
# str(e)
# + "\n\nPlease install Orfeo Toolbox if it isn't installed yet.\n\n"
# "Also, add the otbApplication module path "
# "(usually something like '/usr/lib/otb/python') "
# "to the environment variable PYTHONPATH.\n")
try:
from osgeo import gdal
except ImportError as e:
......@@ -32,81 +41,130 @@ class IdDefaultDict(dict):
return self[key]
#class DataType(object):
# """Abstract class for a data type (int16, int32, float32, etc.)"""
#
# data_type_match = IdDefaultDict()
#
# def __set__(self, instance, value):
# instance.otb_dtype = self.data_type_match[value]
#
# def __get__(self, instance, owner):
# revert_match = {v: k for k, v in self.data_type_match.iteritems()}
# return revert_match[instance.otb_dtype]
class DataType(object):
"""Abstract class for a data type (int16, int32, float32, etc.)"""
data_type_match = IdDefaultDict()
def __set__(self, instance, value):
instance.otb_dtype = self.data_type_match[value]
instance.gdal_dtype = self.data_type_match[value]
def __get__(self, instance, owner):
revert_match = {v: k for k, v in self.data_type_match.iteritems()}
return revert_match[instance.otb_dtype]
return revert_match[instance.gdal_dtype]
class LStrDataType(DataType):
"""Represent a data type given in lower string format (eg. 'int16', 'int32',
'float32', etc.)"""
data_type_match = {'uint8': otb.ImagePixelType_uint8,
'uint16': otb.ImagePixelType_uint16,
'uint32': otb.ImagePixelType_uint32,
'int16': otb.ImagePixelType_int16,
'int32': otb.ImagePixelType_int32,
'float32': otb.ImagePixelType_float,
'float64': otb.ImagePixelType_double}
# data_type_match = {'uint8': otb.ImagePixelType_uint8,
# 'uint16': otb.ImagePixelType_uint16,
# 'uint32': otb.ImagePixelType_uint32,
# 'int16': otb.ImagePixelType_int16,
# 'int32': otb.ImagePixelType_int32,
# 'float32': otb.ImagePixelType_float,
# 'float64': otb.ImagePixelType_double}
data_type_match = {'uint8': gdal.GDT_Byte,
'uint16': gdal.GDT_UInt16,
'uint32': gdal.GDT_UInt32,
'int16': gdal.GDT_Int16,
'int32': gdal.GDT_Int32,
'float32': gdal.GDT_Float32,
'float64': gdal.GDT_Float64}
class UStrDataType(DataType):
"""Represent a data type given in upper string format (eg. 'Int16', 'Int32',
'Float32', etc.)"""
data_type_match = {'UInt8': otb.ImagePixelType_uint8,
'UInt16': otb.ImagePixelType_uint16,
'UInt32': otb.ImagePixelType_uint32,
'Int16': otb.ImagePixelType_int16,
'Int32': otb.ImagePixelType_int32,
'Float32': otb.ImagePixelType_float,
'Float64': otb.ImagePixelType_double}
# data_type_match = {'UInt8': otb.ImagePixelType_uint8,
# 'UInt16': otb.ImagePixelType_uint16,
# 'UInt32': otb.ImagePixelType_uint32,
# 'Int16': otb.ImagePixelType_int16,
# 'Int32': otb.ImagePixelType_int32,
# 'Float32': otb.ImagePixelType_float,
# 'Float64': otb.ImagePixelType_double}
data_type_match = {'UInt8': gdal.GDT_Byte,
'UInt16': gdal.GDT_UInt16,
'UInt32': gdal.GDT_UInt32,
'Int16': gdal.GDT_Int16,
'Int32': gdal.GDT_Int32,
'Float32': gdal.GDT_Float32,
'Float64': gdal.GDT_Float64}
class NumpyDataType(DataType):
"""Represent a data type for Numpy (eg. np.int16, np.int32, np.float32,
etc.)"""
data_type_match = {np.uint8: otb.ImagePixelType_uint8,
np.uint16: otb.ImagePixelType_uint16,
np.uint32: otb.ImagePixelType_uint32,
np.int16: otb.ImagePixelType_int16,
np.int32: otb.ImagePixelType_int32,
np.int64: otb.ImagePixelType_int32,
np.float32: otb.ImagePixelType_float,
np.float64: otb.ImagePixelType_double}
# data_type_match = {np.uint8: otb.ImagePixelType_uint8,
# np.uint16: otb.ImagePixelType_uint16,
# np.uint32: otb.ImagePixelType_uint32,
# np.int16: otb.ImagePixelType_int16,
# np.int32: otb.ImagePixelType_int32,
# np.int64: otb.ImagePixelType_int32,
# np.float32: otb.ImagePixelType_float,
# np.float64: otb.ImagePixelType_double}
data_type_match = {np.uint8: gdal.GDT_Byte,
np.uint16: gdal.GDT_UInt16,
np.uint32: gdal.GDT_UInt32,
np.int16: gdal.GDT_Int16,
np.int32: gdal.GDT_Int32,
np.float32: gdal.GDT_Float32,
np.float64: gdal.GDT_Float64}
class GdalDataType(DataType):
"""Represent a data type for gdal (eg. gdal.GDT_Int16, gdal.GDT_Iint32,
gdal.GDT_Float32, etc.)"""
data_type_match = {gdal.GDT_Byte: otb.ImagePixelType_uint8,
gdal.GDT_UInt16: otb.ImagePixelType_uint16,
gdal.GDT_UInt32: otb.ImagePixelType_uint32,
gdal.GDT_Int16: otb.ImagePixelType_int16,
gdal.GDT_Int32: otb.ImagePixelType_int32,
gdal.GDT_Float32: otb.ImagePixelType_float,
gdal.GDT_Float64: otb.ImagePixelType_double}
# data_type_match = {gdal.GDT_Byte: otb.ImagePixelType_uint8,
# gdal.GDT_UInt16: otb.ImagePixelType_uint16,
# gdal.GDT_UInt32: otb.ImagePixelType_uint32,
# gdal.GDT_Int16: otb.ImagePixelType_int16,
# gdal.GDT_Int32: otb.ImagePixelType_int32,
# gdal.GDT_Float32: otb.ImagePixelType_float,
# gdal.GDT_Float64: otb.ImagePixelType_double}
def __set__(self, instance, value):
instance._gdal_type = value
def __get__(self, instance, owner):
return instance._gdal_type
class OtbDataType(DataType):
"""Represent a data type for orfeo-toolbox
(eg. otb.ImagePixelType_int16)"""
def __set__(self, instance, value):
instance._otb_type = value
def __get__(self, instance, owner):
return instance._otb_type
# def __set__(self, instance, value):
# instance._otb_type = value
#
# def __get__(self, instance, owner):
# return instance._otb_type
if OTB_IS_LOAD:
data_type_match = {otb.ImagePixelType_uint8: gdal.GDT_Byte,
otb.ImagePixelType_uint16: gdal.GDT_UInt16,
otb.ImagePixelType_uint32: gdal.GDT_UInt32,
otb.ImagePixelType_int16: gdal.GDT_Int16,
otb.ImagePixelType_int32: gdal.GDT_Int32,
otb.ImagePixelType_float: gdal.GDT_Float32,
otb.ImagePixelType_double: gdal.GDT_Float64}
else:
def __get__(self, instance, owner):
raise ERROR_OTB
class RasterDataType(object):
......@@ -124,7 +182,7 @@ class RasterDataType(object):
numpy_dtype=None,
otb_dtype=None,
gdal_dtype=None):
if lstr_dtype:
self.lstr_dtype = lstr_dtype
elif ustr_dtype:
......
ymraster/ymraster.py
+ 307
− 65
View file @ c6790093
......@@ -6,22 +6,29 @@ raster images.
try:
import otbApplication as otb
OTB_IS_LOAD = True
try:
app = otb.Registry.CreateApplication('Smoothing')
app.SetParameterString('out', 'foo.tif')
except AttributeError:
OTB_IS_LOAD = False
ERROR_OTB = AttributeError(
"This functionnality is not available without Orfeo Toolbox : "
"Unable to create otbApplication objects\n\n"
"Please set the ITK_AUTOLOAD_PATH environment variable "
"to the Orfeo Toolbox applications folder path "
"(usually something like '/usr/lib/otb/applications') ")
#raise ImportError(MSG_ERROR_OTB)
except ImportError as e:
raise ImportError(
str(e)
+ "\n\nPlease install Orfeo Toolbox if it isn't installed yet.\n\n"
OTB_IS_LOAD = False
ERROR_OTB = ImportError(str(e)
+ "\n\nThis functionnality is not available without Orfeo Toolbox. " +
"Please install Orfeo Toolbox if it isn't installed yet.\n\n"
"Also, add the otbApplication module path "
"(usually something like '/usr/lib/otb/python') "
"to the PYTHONPATH environment variable.")
try:
app = otb.Registry.CreateApplication('Smoothing')
app.SetParameterString('out', 'foo.tif')
except AttributeError:
raise ImportError(
"Unable to create otbApplication objects\n\n"
"Please set the ITK_AUTOLOAD_PATH environment variable "
"to the Orfeo Toolbox applications folder path "
"(usually something like '/usr/lib/otb/applications') ")
#raise ImportError(MSG_ERROR)
try:
from osgeo import osr, gdal
......@@ -49,6 +56,16 @@ from time import mktime
import os
import shutil
from tempfile import gettempdir
from functools import wraps
def otb_function(f):
@wraps(f)
def wrapper(*args, **kwds):
if not OTB_IS_LOAD:
raise ERROR_OTB
return f(*args, **kwds)
return wrapper
def add_suffix(path_file, suffix):
"""
......@@ -95,53 +112,59 @@ def write_file(out_filename, array=None, overwrite=True,
"""Writes a NumPy array to an image file.
If there is no array (array is None), the function simply create an empty
image file of given size (width, height, depth). In other words, if array is
None, width, height and depth must be specified.
image file of given size (width, height, depth). In other words, if array
is None, width, height and depth must be specified.
If array is not None, then all other parameters are optional: if the file
exists, array will be written into the file. If the file does not exists, a
new file will be created with same size and type than the array.
exists, array will be written into the file. If the file does not exists,
a new file will be created with same size and type than the array.
This allows to write a file block by block. First create an empty file of
correct size and type. Then write each block into the file using the xoffset
and yoffset parameters.
correct size and type. Then write each block into the file using the
xoffset and yoffset parameters.
:param out_filename: path to the output file
:type out_filename: str
:param array: the NumPy array to save. None means that an empty file will be
created or, if file exists, that no data will be written
(except metadata if specified)
:type array: np.ndarray
:param overwrite: if True, overwrite file if exists. If False, open the
file, then update it or create it if it does not exist.
True by default.
:type overwrite: bool
:param xoffset: horizontal offset. First index from which to write the array
in the output file if the array is smaller (default: 0)
:type xoffset: float
:param yoffset: Vertical offset. First index from which to write the array
in the output file if the array is smaller (default: 0)
:type yoffset: float
:param band_idx: depth offset. First band from which to write the array
in the output file if the array is smaller (default: 1)
:type band_idx: int
:param width: horizontal size of the image to be created
:type width: int
:param height: vertical size of the image to be created
:type width: int
:param depth: number of bands of the image to be created
:type depth: int
:param dtype: data type to use for the output file. None means that the file
already exists
:type dtype: RasterDataType
:param date_time: date/time to write in the output file metadata
:type date_time: datetime.datetime
:param nodata_value: value to set as NODATA in the output file metadata
:type nodata_value: dtype
:param srs: projection to write in the output file metadata
:type srs: osr.SpatialReference
:param transform: geo-transformation to use for the output file
:type transform: 6-tuple of floats
Parameters
----------
out_filename : str,
Path to the output file.
array : np.ndarray,
The NumPy array to save. None means that an empty file will be
created or, if file exists, that no data will be written (except
metadata if specified)
overwrite : bool,
if True, overwrite file if exists. If False, open the
file, then update it or create it if it does not exist. True by
default.
xoffset : flaot,
Horizontal offset. First index from which to write the array
in the output file if the array is smaller (default: 0).
yoffset : float,
Vertical offset. First index from which to write the array
in the output file if the array is smaller (default: 0).
band_idx : int
Depth offset. First band from which to write the array
in the output file if the array is smaller (default: 1).
kw
--
width : int,
Horizontal size of the image to be created (in pixel)
height : int,
Vertical size of the image to be created (in pixel)
depth : int,
Number of bands of the image to be created.
dtype : RasterDataType
Data type to use for the output file. None means that the file
already exists.
date_time : datetime.datetime
Date/time to write in the output file metadata.
nodata_value : dtype
Value to set as NODATA in the output file metadata.
srs : osr.SpatialReference
projection to write in the output file metadata.
transform : 6-tuple of floats
Geo-transformation to use for the output file.
band_info : list of string:
Info of each band to write in the output file metadata
"""
# Size & data type of output image
xsize, ysize = (kw['width'], kw['height']) \
......@@ -173,7 +196,8 @@ def write_file(out_filename, array=None, overwrite=True,
# Set metadata
if kw.get('date_time'):
out_ds.SetMetadata(
{'TIFFTAG_DATETIME': kw['date_time'].strftime('%Y:%m:%d %H:%M:%S')})
{'TIFFTAG_DATETIME':
kw['date_time'].strftime('%Y:%m:%d %H:%M:%S')})
if kw.get('srs'):
out_ds.SetProjection(kw['srs'].ExportToWkt())
if kw.get('transform'):
......@@ -188,6 +212,10 @@ def write_file(out_filename, array=None, overwrite=True,
band.WriteArray(array, xoff=xoffset, yoff=yoffset)
if kw.get('nodata_value'):
band.SetNoDataValue(kw.get('nodata_value'))
if kw.get('band_info'):
metadata = band.GetMetadata()
metadata['band_info'] = kw.get('band_info')[0]
band.SetMetadata(metadata)
band.FlushCache()
else:
for i in range(number_bands):
......@@ -195,11 +223,15 @@ def write_file(out_filename, array=None, overwrite=True,
band.WriteArray(array[:, :, i], xoff=xoffset, yoff=yoffset)
if kw.get('nodata_value'):
band.SetNoDataValue(kw.get('nodata_value'))
if kw.get('band_info'):
metadata = band.GetMetadata()
metadata['band_info'] = kw.get('band_info')[i]
band.SetMetadata(metadata)
band.FlushCache()
band = None
out_ds = None
@otb_function
def concatenate_rasters_old(*rasters, **kw):
"""Write a raster which is the concatenation of the given rasters, in order.
......@@ -256,6 +288,8 @@ def concatenate_rasters_old(*rasters, **kw):
shutil.copy(out_filename, raster0.filename)
os.remove(out_filename)
@otb_function
def concatenate_rasters(rasters, **kw):
"""Write a raster which is the concatenation of the given rasters, in order.
......@@ -637,6 +671,125 @@ class Raster(Sized):
# Close file
ds = None
def write_metadata(self, metadata):
"""
Write metada in the file
Parameters
----------
metadata : dictionnary
Key are Tag of the metadata, values is the corresponding
description
Limitations
-----------
Works for Tif files only.
"""
ds = gdal.Open(self._filename, gdal.GA_Update)
metadata_orign = ds.GetMetadata()
metadata_orign.update(metadata)
ds.SetMetadata(metadata_orign)
ds = None
self.refresh()
def write_band_metadata(self, idx, metadata):
"""
Write band metada in the file
Parameters
----------
idx : int
Band number. Indexation starts at 1.
metadata : dictionnary
Key are Tag of the metadata, values is the corresponding
description
Limitations
-----------
Works for Tif files only.
"""
ds = gdal.Open(self._filename, gdal.GA_Update)
band = ds.GetRasterBand(idx)
metadata_orign = band.GetMetadata()
metadata_orign.update(metadata)
band.SetMetadata(metadata)
ds = None
self.refresh()
def get_band_metadata(self, idx):
"""
Get band metada in the file
Parameters
----------
idx : int
Band number. Indexation starts at 1.
Returns
-------
metadata : dictionnary
Key are Tag of the metadata, values is the corresponding
description
"""
ds = gdal.Open(self._filename, gdal.GA_Update)
band = ds.GetRasterBand(idx)
metadata = band.GetMetadata()
ds = None
return metadata
@otb_function
def change_dtype(self, out_dtype, out_filename):
"""
Convert Raster datatype to an other using Otb BandMath application
Parameters
----------
dytpe : str
Can be one of this type :
- 'uint8'
- 'uint16'
- 'uint32'
- 'int16'
- 'int32'
- 'float32'
- 'float64'
Returns
-------
out_rst : `Raster`
Output raster
Limitation
----------
Have not been tested in case rescaling and.or resampling is needed
"""
dtype = RasterDataType(lstr_dtype=out_dtype).otb_dtype
expr = ''
for i in range(self.count):
expr_b = 'im1b{}'.format(i + 1)
if i + 1 < self.count:
expr_b += ';'
expr += expr_b
print 'BandMath expression is : ', expr
# Apply bandMathx application
BandMath = otb.Registry.CreateApplication("BandMathX")
BandMath.SetParameterStringList("il", [self.filename])
BandMath.SetParameterString("out", out_filename)
BandMath.SetParameterString("exp", expr)
BandMath.SetParameterOutputImagePixelType("out", dtype)
BandMath.ExecuteAndWriteOutput()
# write new nodata value in metadata
out_rst = Raster(out_filename)
return out_rst
def band_statistics(self, idx):
"""
Compute band statistic, without taking into account nodata values
......@@ -663,6 +816,72 @@ class Raster(Sized):
# False for approximation
return band.ComputeStatistics(False)
def compute_percentile(self, list_per, list_idx=None,):
"""
Compute and return percentile(s) of Raster's bands and write it in
bands' metadata.
Parameters
---------
list_per : list of int.
Percentiles must be comprised between 0 and 100
list_idx: list of int (optional)
List of band index (idexation starts at 1) on which compute
percentile(s). All band are processed by default.
Returns
-------
metadata_all_bands : dict of dict:
Dictionnary of metadatas for each band.
"""
metadata_all_bands = {}
ds = gdal.Open(self.filename, gdal.GA_ReadOnly)
if list_idx is None:
list_idx = range(1, self.count + 1)
for idx in list_idx:
tab = self.array_from_bands(idx, mask_nodata=True)
tab = tab.compressed()
band = ds.GetRasterBand(idx)
metadata = band.GetMetadata()
for per in list_per:
per_label = 'STATISTICS_PER_{}'.format(per)
per_value = np.nanpercentile(tab, per)
metadata[per_label] = str(per_value)
metadata_all_bands[idx] = metadata
band.SetMetadata(metadata)
return metadata_all_bands
def get_percentile(self, per, idx):
"""
Return percentile of a Raster's band according to its metadata.
Parameters
---------
per : int.
Percentile, must be comprised between 0 and 100.
idx: int
Band index (idexation starts at 1) on which compute
percentile `per`.
Returns
-------
percentile_value : float
Percentile value
See also
--------
Raster.compute_percentile()
"""
ds = gdal.Open(self.filename, gdal.GA_ReadOnly)
band = ds.GetRasterBand(idx)
per_label = 'STATISTICS_PER_{}'.format(per)
return float(band.GetMetadata()[per_label])
def get_projection_from(self, raster):
"""TODO
"""
......@@ -850,6 +1069,7 @@ class Raster(Sized):
block_win[0],
block_win[1])
@otb_function
def remove_bands(self, *idxs, **kw):
"""
Saves a new raster with the specified bands removed.
......@@ -906,6 +1126,7 @@ class Raster(Sized):
else:
return Raster(out_filename)
@otb_function
def extract_band(self,*idxs,**kw):
"""
Saves a new raster with the specified bands extracted and the others
......@@ -938,18 +1159,19 @@ class Raster(Sized):
SplitImage.ExecuteAndWriteOutput()
# Out file
out_filename = kw.get('out_filename',os.path.join(gettempdir(), 'bands_removed.tif'))
out_filename = kw.get('out_filename',
os.path.join(gettempdir(), 'bands_removed.tif'))
# Concatenate the mono-band images with only the wanted band(s)
#------------------------------------------------------------------
list_path = [os.path.join(gettempdir(), 'splitted_{}.tif'.format(i))
for i in range(self._count)
if i + 1 in indices]
list_path = [os.path.join(gettempdir(),
'splitted_{}.tif'.format(i - 1))
for i in indices]
#Re order if indicated
temp_list = [(idx, path) for idx, path in zip(indices,list_path)]
temp_list.sort()
list_path = [path for (idx, path) in temp_list]
# temp_list = [(idx, path) for idx, path in zip(indices,list_path)]
# temp_list.sort()
# list_path = [path for (idx, path) in temp_list]
#concatenate here
ConcatenateImages = otb.Registry.CreateApplication("ConcatenateImages")
......@@ -971,6 +1193,7 @@ class Raster(Sized):
else:
return Raster(out_filename)
@otb_function
def rescale_bands(self, dstmin, dstmax, nodata_dst=None, out_filename=None,
dtype=None,):
"""
......@@ -1096,6 +1319,7 @@ class Raster(Sized):
else:
return Raster(out_filename)
@otb_function
def normalize(self, nodata_dst=None, out_filename=None,
dtype=None):
"""
......@@ -1157,8 +1381,14 @@ class Raster(Sized):
BandMath.SetParameterOutputImagePixelType("out", dtype.otb_dtype)
BandMath.ExecuteAndWriteOutput()
return Raster(out_filename)
# write new nodata value in metadata
out_rst = Raster(out_filename)
if nodata_dst:
out_rst.nodata_value = nodata_dst
return out_rst
@otb_function
def fusion(self, pan, **kw):
"""
Sharpen the raster with its corresponding panchromatic image.
......@@ -1208,6 +1438,7 @@ class Raster(Sized):
else:
return Raster(out_filename)
@otb_function
@fix_missing_proj
def radiometric_indices(self, *indices, **kw):
"""Saves a raster of radiometric indices about the raster.
......@@ -1281,6 +1512,7 @@ class Raster(Sized):
out_raster.date_time = self._date_time
return out_raster
@otb_function
def ndvi(self, red_idx, nir_idx, **kw):
"""Saves the Normalized Difference Vegetation Index (NDVI) of the
raster.
......@@ -1308,6 +1540,7 @@ class Raster(Sized):
nir_idx=nir_idx,
out_filename=out_filename)
@otb_function
def ndwi(self, nir_idx, mir_idx, **kw):
"""Saves the Normalized Difference Water Index (NDWI) of the raster.
......@@ -1334,6 +1567,7 @@ class Raster(Sized):
mir_idx=mir_idx,
out_filename=out_filename)
@otb_function
def mndwi(self, green_idx, mir_idx, **kw):
"""Saves the Modified Normalized Difference Water Index (MNDWI) of the
raster.
......@@ -1363,6 +1597,7 @@ class Raster(Sized):
ndsi = mndwi
@otb_function
def append(self, *rasters):
"""Append the given rasters into the current one.
......@@ -1374,6 +1609,7 @@ class Raster(Sized):
raster_list = [self] + list(rasters)
concatenate_rasters(raster_list)
@otb_function
def apply_mask(self, mask_raster, mask_value=1, set_value=None, **kw):
"""Apply a mask to the raster: set all masked pixels to a given value.
......@@ -1442,7 +1678,7 @@ class Raster(Sized):
masked_bands = [Raster(os.path.join(gettempdir(),
'mono_masked_{}.tif'.format(i)))
for i in range(self._count)]
concatenate_rasters(*masked_bands, out_filename=out_filename)
concatenate_rasters(masked_bands, out_filename=out_filename)
out_raster = Raster(out_filename)
out_raster.nodata_value = set_value
print set_value
......@@ -1453,6 +1689,7 @@ class Raster(Sized):
return out_raster
@otb_function
def change_value(self, dict_change, idx_band, **kw):
"""
Change values by others in raster band.
......@@ -1538,6 +1775,7 @@ class Raster(Sized):
return list_long, list_lat
@otb_function
def _lsms_smoothing(self, spatialr, ranger, thres=0.1, rangeramp=0,
maxiter=10, **kw):
"""First step in a Large-Scale Mean-Shift (LSMS) segmentation: smooths
......@@ -1599,6 +1837,7 @@ class Raster(Sized):
return Raster(out_filename), Raster(out_spatial_filename)
@otb_function
def _lsms_segmentation(self, spatialr, ranger, spatial_raster,
block_size=None, **kw):
"""Second step in a LSMS segmentation: performs the actual object
......@@ -1669,6 +1908,7 @@ class Raster(Sized):
return Raster(out_filename)
@otb_function
@fix_missing_proj
def _lsms_merging(self, object_minsize, smoothed_raster, block_size=None,
**kw):
......@@ -1735,6 +1975,7 @@ class Raster(Sized):
else:
return Raster(out_filename)
@otb_function
def _lsms_vectorization(self, orig_raster, block_size=None, **kw):
"""Fourth and Last (optional) step in a LSMS segmentation: vectorize a
labeled segmented image, turn each object into a polygon. Each polygon
......@@ -1785,6 +2026,7 @@ class Raster(Sized):
LSMSVectorization.SetParameterInt("tilesizey", tilesizey)
LSMSVectorization.ExecuteAndWriteOutput()
@otb_function
def lsms_segmentation(self,
spatialr,
ranger,
......
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