From 9e012a2dca9bef0dab6855b20b37b794ad2891a2 Mon Sep 17 00:00:00 2001
From: Julien Michel <julien.michel@cnes.fr>
Date: Fri, 30 Oct 2015 10:43:46 +0100
Subject: [PATCH] ENH: Avoid possible division by zero
---
Modules/Learning/Supervised/include/otbCvRTreesWrapper.h | 6 ++++++
1 file changed, 6 insertions(+)
diff --git a/Modules/Learning/Supervised/include/otbCvRTreesWrapper.h b/Modules/Learning/Supervised/include/otbCvRTreesWrapper.h
index 205d4e7cc2..290bd6af32 100644
--- a/Modules/Learning/Supervised/include/otbCvRTreesWrapper.h
+++ b/Modules/Learning/Supervised/include/otbCvRTreesWrapper.h
@@ -47,6 +47,12 @@ public:
const cv::Mat& missing =
cv::Mat()) const
{
+ // Sanity check (division by ntrees later on)
+ if(ntrees == 0)
+ {
+ return 0.;
+ }
+
std::vector<unsigned int> classVotes(nclasses);
for( int k = 0; k < ntrees; k++ )
{
--
GitLab