Commit 36a2d1a6 authored by Remi Cresson's avatar Remi Cresson
Browse files

TEST: inference no-data test

parent 5f31067a
2 merge requests!88Release v4.1.0,!83Tfmodelserve nodata
Pipeline #46064 passed with stages
in 18 minutes and 20 seconds
Showing with 9 additions and 2 deletions
+9 -2
...@@ -20,7 +20,7 @@ class NodataInferenceTest(unittest.TestCase): ...@@ -20,7 +20,7 @@ class NodataInferenceTest(unittest.TestCase):
# OTB pipeline # OTB pipeline
bmx = otbApplication.Registry.CreateApplication("BandMathX") bmx = otbApplication.Registry.CreateApplication("BandMathX")
bmx.SetParameterString("exp", "{idxX>idxY?1:0}") bmx.SetParameterString("exp", "{idxX>idxY?idxX*idxY:0}")
bmx.SetParameterStringList( bmx.SetParameterStringList(
"il", [resolve_paths("$DATADIR/fake_spot6.jp2")] "il", [resolve_paths("$DATADIR/fake_spot6.jp2")]
) )
...@@ -30,12 +30,19 @@ class NodataInferenceTest(unittest.TestCase): ...@@ -30,12 +30,19 @@ class NodataInferenceTest(unittest.TestCase):
"TensorflowModelServe" "TensorflowModelServe"
) )
infer.SetParameterString("model.dir", sm_dir) infer.SetParameterString("model.dir", sm_dir)
infer.SetParameterString("model.fullyconv", "on")
infer.AddImageToParameterInputImageList( infer.AddImageToParameterInputImageList(
"source1.il", bmx.GetParameterOutputImage("out") "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.SetParameterString("out", resolve_paths("$TMPDIR/nd_out.tif"))
infer.ExecuteAndWriteOutput() infer.ExecuteAndWriteOutput()
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() NodataInferenceTest().test_infersimple()
#unittest.main()
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment