fix: typos

luxonis · Nov 9, 2024 · 1ed14ba · 1ed14ba
1 parent 7b999ac
commit 1ed14ba
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/depthai_nodes/ml/parsers/yunet.py b/depthai_nodes/ml/parsers/yunet.py
@@ -133,13 +133,12 @@ def build(
         """
 
         super().build(head_config)
-        output_layers = head_config.get("outputs", [])
         if len(inputs_size) != 1:
             raise ValueError(
                 f"Only one input supported for YuNetParser, got {len(inputs_size)} inputs."
             )
         self.input_size = inputs_size[0]
-        print(self.input_size)
+        output_layers = head_config.get("outputs", [])
         for output_layer in output_layers:
             if "loc" in output_layer:
                 self.loc_output_layer_name = output_layer

diff --git a/depthai_nodes/parser_generator.py b/depthai_nodes/parser_generator.py
@@ -58,7 +58,6 @@ def build(self, nn_archive: dai.NNArchive, head_index: int = None) -> Dict:
             if "inputs_size" in sig.parameters:
                 inputs_size = []
                 for input in nn_archive.getConfig().model.inputs:
-                    breakpoint()
                     if input.layout == "NHWC":
                         _, height, width, _ = input.shape
                     elif input.layout == "NCHW":