From a18cf8c6341df534a8d1b7cd68168825ce1e40d5 Mon Sep 17 00:00:00 2001
From: Jordi Inglada <jordi.inglada@cesbio.cnes.fr>
Date: Thu, 1 Mar 2018 15:11:29 +0100
Subject: [PATCH] ENH: find KNN in parallel for smote
---
Modules/Learning/Sampling/include/otbSampleAugmentation.h | 1 +
1 file changed, 1 insertion(+)
diff --git a/Modules/Learning/Sampling/include/otbSampleAugmentation.h b/Modules/Learning/Sampling/include/otbSampleAugmentation.h
index b778e79f26..a64e663e28 100644
--- a/Modules/Learning/Sampling/include/otbSampleAugmentation.h
+++ b/Modules/Learning/Sampling/include/otbSampleAugmentation.h
@@ -147,6 +147,7 @@ void findKNNIndices(const SampleVectorType& inSamples,
{
const auto nbSamples = inSamples.size();
nnVector.resize(nbSamples);
+ #pragma omp parallel for
for(size_t sampleIdx=0; sampleIdx<nbSamples; ++sampleIdx)
{
NNIndicesType nns;
--
GitLab