|
@@ -25,6 +25,7 @@ int main ( int argc, char **argv )
|
|
|
NICE::Config conf (argc, argv );
|
|
|
NICE::ResourceStatistics rs;
|
|
|
|
|
|
+ bool postProcessing = conf.gB( "SemSegObliqueTree", "post_process", false);
|
|
|
MultiDataset md ( &conf );
|
|
|
const ClassNames & classNames = md.getClassNames ( "train" );
|
|
|
const LabeledSet *testFiles = md["test"];
|
|
@@ -86,9 +87,12 @@ int main ( int argc, char **argv )
|
|
|
<< "\n\n";
|
|
|
|
|
|
// post processing results
|
|
|
- NICE::Image postIm(segresult.width(), segresult.height());
|
|
|
- NICE::median(segresult, &postIm, 1);
|
|
|
- segresult = postIm;
|
|
|
+ if (postProcessing)
|
|
|
+ {
|
|
|
+ NICE::Image postIm(segresult.width(), segresult.height());
|
|
|
+ NICE::median(segresult, &postIm, 1);
|
|
|
+ segresult = postIm;
|
|
|
+ }
|
|
|
|
|
|
// updating confusion matrix
|
|
|
SemSegTools::updateConfusionMatrix (
|