From c95ae4175be67dd66769d6a102a021211bc88abe Mon Sep 17 00:00:00 2001
From: Guillaume Pasero <guillaume.pasero@c-s.fr>
Date: Tue, 28 Jul 2015 14:20:31 +0200
Subject: [PATCH] ENH: update classification appli to output confidence mask
---
.../AppClassification/app/otbImageClassifier.cxx | 15 +++++++++++++++
1 file changed, 15 insertions(+)
diff --git a/Modules/Applications/AppClassification/app/otbImageClassifier.cxx b/Modules/Applications/AppClassification/app/otbImageClassifier.cxx
index 07a8a352d7..8f7b676fb0 100644
--- a/Modules/Applications/AppClassification/app/otbImageClassifier.cxx
+++ b/Modules/Applications/AppClassification/app/otbImageClassifier.cxx
@@ -59,6 +59,7 @@ public:
typedef ClassificationFilterType::ValueType ValueType;
typedef ClassificationFilterType::LabelType LabelType;
typedef otb::MachineLearningModelFactory<ValueType, LabelType> MachineLearningModelFactoryType;
+ typedef ClassificationFilterType::ConfidenceImageType ConfidenceImageType;
private:
void DoInit()
@@ -94,6 +95,11 @@ private:
SetParameterDescription( "out", "Output image containing class labels");
SetParameterOutputImagePixelType( "out", ImagePixelType_uint8);
+ AddParameter(ParameterType_OutputImage, "confmap", "Confidence map");
+ SetParameterDescription( "confmap", "Confidence map of the produced classification. Meaning depends on the model.");
+ SetParameterOutputImagePixelType( "confmap", ImagePixelType_double);
+ MandatoryOff("confmap");
+
AddRAMParameter();
// Doc example parameter settings
@@ -171,6 +177,15 @@ private:
}
SetParameterOutputImage<OutputImageType>("out", m_ClassificationFilter->GetOutput());
+
+ // output confidence map
+ if (IsParameterEnabled("confmap") && HasValue("confmap"))
+ {
+ if (m_Model->HasConfidenceIndex())
+ {
+ SetParameterOutputImage<ConfidenceImageType>("confmap",m_ClassificationFilter->GetOutputConfidence());
+ }
+ }
}
ClassificationFilterType::Pointer m_ClassificationFilter;
--
GitLab