From 36a2d1a6b299d7a4658b5ce526f9f1ae96c79e4c Mon Sep 17 00:00:00 2001 From: Remi Cresson <remi.cresson@inrae.fr> Date: Thu, 6 Apr 2023 11:29:29 +0200 Subject: [PATCH] TEST: inference no-data test --- test/nodata_test.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/test/nodata_test.py b/test/nodata_test.py index 0f106af..93e165b 100644 --- a/test/nodata_test.py +++ b/test/nodata_test.py @@ -20,7 +20,7 @@ class NodataInferenceTest(unittest.TestCase): # OTB pipeline bmx = otbApplication.Registry.CreateApplication("BandMathX") - bmx.SetParameterString("exp", "{idxX>idxY?1:0}") + bmx.SetParameterString("exp", "{idxX>idxY?idxX*idxY:0}") bmx.SetParameterStringList( "il", [resolve_paths("$DATADIR/fake_spot6.jp2")] ) @@ -30,12 +30,19 @@ class NodataInferenceTest(unittest.TestCase): "TensorflowModelServe" ) infer.SetParameterString("model.dir", sm_dir) + infer.SetParameterString("model.fullyconv", "on") infer.AddImageToParameterInputImageList( "source1.il", bmx.GetParameterOutputImage("out") ) + infer.SetParameterFloat("source1.nodata", 0.0) + for param in [ + "source1.rfieldx", "source1.rfieldy", "output.efieldx", "output.efieldy" + ]: + infer.SetParameterInt(param, 16) infer.SetParameterString("out", resolve_paths("$TMPDIR/nd_out.tif")) infer.ExecuteAndWriteOutput() if __name__ == '__main__': - unittest.main() + NodataInferenceTest().test_infersimple() + #unittest.main() -- GitLab