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++){