Commit cf51ccb3 authored by Cédric Traizet's avatar Cédric Traizet
Browse files

ENH: input centroid file in the composite application

No related merge requests found
Showing with 16 additions and 5 deletions
+16 -5
...@@ -81,9 +81,13 @@ protected: ...@@ -81,9 +81,13 @@ protected:
SetDefaultParameterInt("maxit", 1000); SetDefaultParameterInt("maxit", 1000);
MandatoryOff("maxit"); MandatoryOff("maxit");
AddParameter(ParameterType_String, "incentroid", "Maximum number of iterations"); AddParameter(ParameterType_String, "inmeans", "Maximum number of iterations");
SetParameterDescription("incentroid", "Maximum number of iterations for the learning step."); SetParameterDescription("inmeans", "Maximum number of iterations for the learning step.");
MandatoryOff("incentroid"); MandatoryOff("inmeans");
AddParameter(ParameterType_Bool, "normalizeinmeans", "Number of classes");
SetParameterDescription("normalizeinmeans", "Number of modes, which will be used to generate class membership.");
SetDefaultParameterInt("normalizeinmeans", true);
AddParameter(ParameterType_OutputFilename, "outmeans", "Centroid filename"); AddParameter(ParameterType_OutputFilename, "outmeans", "Centroid filename");
SetParameterDescription("outmeans", "Output text file containing centroid positions"); SetParameterDescription("outmeans", "Output text file containing centroid positions");
...@@ -252,8 +256,15 @@ protected: ...@@ -252,8 +256,15 @@ protected:
GetParameterInt("maxit")); GetParameterInt("maxit"));
GetInternalApplication("training")->SetParameterInt("classifier.sharkkm.k", GetInternalApplication("training")->SetParameterInt("classifier.sharkkm.k",
GetParameterInt("nc")); GetParameterInt("nc"));
GetInternalApplication("training")->SetParameterString("classifier.sharkkm.incentroid", if(IsParameterEnabled("inmeans") && HasValue("inmeans"))
GetParameterString("incentroid")); {
GetInternalApplication("training")->SetParameterString("classifier.sharkkm.centroids",
GetParameterString("inmeans"));
if(GetParameterInt("normalizeinmeans"))
GetInternalApplication("training")->SetParameterString("classifier.sharkkm.centroidstats",
GetInternalApplication("imgstats")->GetParameterString("out"));
}
if( IsParameterEnabled("rand")) if( IsParameterEnabled("rand"))
GetInternalApplication("training")->SetParameterInt("rand", GetParameterInt("rand")); GetInternalApplication("training")->SetParameterInt("rand", GetParameterInt("rand"));
......
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