fix input shape parsing (#83)

micromind-toolkit · Jan 16, 2024 · ef961a0 · ef961a0
1 parent e9af6c5
commit ef961a0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/recipes/object_detection/inference.py b/recipes/object_detection/inference.py
@@ -62,7 +62,7 @@ def forward(self, img):
     )
 
     hparams = parse_configuration(sys.argv[1])
-    if isinstance(hparams.input_shape, str):
+    if len(hparams.input_shape) != 3:
         hparams.input_shape = [
             int(x) for x in "".join(hparams.input_shape).split(",")
         ]  # temp solution

diff --git a/recipes/object_detection/train.py b/recipes/object_detection/train.py
@@ -231,7 +231,7 @@ def replace_datafolder(hparams, data_cfg):
 if __name__ == "__main__":
     assert len(sys.argv) > 1, "Please pass the configuration file to the script."
     hparams = parse_configuration(sys.argv[1])
-    if isinstance(hparams.input_shape, str):
+    if len(hparams.input_shape) != 3:
         hparams.input_shape = [
             int(x) for x in "".join(hparams.input_shape).split(",")
         ]  # temp solution