def kmeans(layer, working_directory, nb_class=None):
"""
WARNING: nb_valid_pixels à calculer ?
"""
internal_working_directory = os.path.join(working_directory, "Internal")
if not os.path.exists(internal_working_directory):
os.makedirs(internal_working_directory)
logger.debug("enntree dans le kmeans")
bands = []
if nb_class is None:
testqt, ok = QInputDialog.getInt(None, "Kmeans", "Nombre de classes", 5)
if ok:
nb_class = testqt
# mask = OTBApplications.bandmath([layer.get_source()], "if(im1b1>0,1,0)", working_directory, "mask")
output = OTBApplications.kmeans_cli(layer.get_source(), nb_class, internal_working_directory)
image_ref = recompose_image(layer, internal_working_directory)
# if not os.path.isfile(output_colored):
output_colored = OTBApplications.color_mapping_cli_ref_image(output, image_ref, working_directory)
return output_colored
def kmeans(layer, working_directory, nb_class=None):
"""
WARNING: nb_valid_pixels à calculer ?
"""
internal_working_directory = os.path.join(working_directory, "Internal")
if not os.path.exists(internal_working_directory):
os.makedirs(internal_working_directory)
logger.debug("enntree dans le kmeans")
bands = []
if nb_class is None:
testqt, ok = QInputDialog.getInt(None, "Kmeans", "Nombre de classes",
5)
if ok:
nb_class = testqt
# mask = OTBApplications.bandmath([layer.get_source()], "if(im1b1>0,1,0)", working_directory, "mask")
output = OTBApplications.kmeans_cli(layer.get_source(), nb_class,
internal_working_directory)
image_ref = recompose_image(layer, internal_working_directory)
# if not os.path.isfile(output_colored):
output_colored = OTBApplications.color_mapping_cli_ref_image(
output, image_ref, working_directory)
return output_colored
