|
@@ -297,8 +297,9 @@ void CodebookRandomForest::pruneForest ()
|
|
|
i++;
|
|
|
}
|
|
|
}
|
|
|
-
|
|
|
+#ifdef DEBUGPRUNING
|
|
|
fprintf (stderr, "Final number of leafs: %ld (%d)\n", leafs, restrictedCodebookSize );
|
|
|
+#endif
|
|
|
}
|
|
|
|
|
|
void CodebookRandomForest::buildLeafMap ()
|
|
@@ -334,8 +335,9 @@ void CodebookRandomForest::buildLeafMap ()
|
|
|
DecisionNode *node = i->second;
|
|
|
leafMap.insert ( pair<DecisionNode *, int> ( node, leafMap.size() ) );
|
|
|
}
|
|
|
-
|
|
|
+#ifdef DEBUGPRUNING
|
|
|
fprintf (stderr, "CSRandomForest::buildLeafMap: dimension = %d\n", (int)leafMap.size() );
|
|
|
+#endif
|
|
|
|
|
|
reinit ( leafMap.size() );
|
|
|
}
|