FIX: inference wrt OTBTF 3.3.0

7762e2b9 · Remi Cresson · 65aecc47 · 7762e2b9 · 7762e2b9 · 7762e2b9
Commit 7762e2b9 authored 2 years ago by Remi Cresson
Hide whitespace changes
Inline Side-by-side

Showing

with 3 additions and 3 deletions
+3 -3
--- a/decloud/production/inference.py
+++ b/decloud/production/inference.py
@@ -86,7 +86,7 @@ def inference(sources, sources_scales, pad, ts, savedmodel_dir, out_tensor, out_
                           get_key("placeholder"): placeholder})

    # Model
-    parameters.update({"model.dir": savedmodel_dir, "model.fullyconv": 1,
+    parameters.update({"model.dir": savedmodel_dir, "model.fullyconv": True,
                       "output.names": [padded_tensor_name(out_tensor, pad)],
                       "output.efieldx": efield, "output.efieldy": efield,
                       "optim.tilesizex": efield, "optim.tilesizey": efield,

--- a/decloud/production/monthly_synthesis_processor_s2.py
+++ b/decloud/production/monthly_synthesis_processor_s2.py
@@ -109,7 +109,7 @@ def monthly_synthesis_inference(sources, sources_scales, pad, ts, savedmodel_dir
                             get_key("placeholder"): placeholder})

    # Model
-    infer_params.update({"model.dir": savedmodel_dir, "model.fullyconv": 1,
+    infer_params.update({"model.dir": savedmodel_dir, "model.fullyconv": True,
                         "output.names": [padded_tensor_name(out_tensor, pad)],
                         "output.efieldx": efield, "output.efieldy": efield,
                         "optim.tilesizex": efield, "optim.tilesizey": efield,

--- a/decloud/production/monthly_synthesis_processor_s2s1.py
+++ b/decloud/production/monthly_synthesis_processor_s2s1.py
@@ -109,7 +109,7 @@ def monthly_synthesis_inference(sources, sources_scales, pad, ts, savedmodel_dir
                             get_key("placeholder"): placeholder})

    # Model
-    infer_params.update({"model.dir": savedmodel_dir, "model.fullyconv": 1,
+    infer_params.update({"model.dir": savedmodel_dir, "model.fullyconv": True,
                         "output.names": [padded_tensor_name(out_tensor, pad)],
                         "output.efieldx": efield, "output.efieldy": efield,
                         "optim.tilesizex": efield, "optim.tilesizey": efield,