From 4c3f8006d1f897bf420d4586dc0eecf5e1b44c95 Mon Sep 17 00:00:00 2001
From: Julien Malik <julien.malik@c-s.fr>
Date: Thu, 19 Feb 2015 15:27:45 +0100
Subject: [PATCH] TEST: avoid segfault in otbImageClassificationFilter
---
.../LearningBase/test/otbImageClassificationFilter.cxx | 6 ++++++
1 file changed, 6 insertions(+)
diff --git a/Modules/Learning/LearningBase/test/otbImageClassificationFilter.cxx b/Modules/Learning/LearningBase/test/otbImageClassificationFilter.cxx
index f2e37b8ab5..760a4fdb87 100644
--- a/Modules/Learning/LearningBase/test/otbImageClassificationFilter.cxx
+++ b/Modules/Learning/LearningBase/test/otbImageClassificationFilter.cxx
@@ -59,6 +59,12 @@ int otbImageClassificationFilter(int itkNotUsed(argc), char * argv[])
model = MachineLearningModelFactoryType::CreateMachineLearningModel(modelfname,
MachineLearningModelFactoryType::ReadMode);
+ if (model.IsNull())
+ {
+ std::cerr << "Unable to create a model from " << modelfname << std::endl;
+ return EXIT_FAILURE;
+ }
+
model->Load(modelfname);
filter->SetModel(model);
--
GitLab