Fix segfault (#2)

uav4geo · Mar 7, 2023 · 2ccbc76 · 2ccbc76
1 parent cd50192
commit 2ccbc76
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/randomforest.cpp b/randomforest.cpp
@@ -246,6 +246,7 @@ void classify(PointSet &pointSet,
   }
 
   std::size_t correct = 0;
+  if (!pointSet.hasLabels()) pointSet.labels.resize(pointSet.count());
 
   #pragma omp parallel for
   for (size_t i = 0; i < pointSet.count(); i++){