fix: ImageOutputParser dequantization

luxonis · Jul 19, 2024 · 29c8718 · 29c8718
1 parent 79b0675
commit 29c8718
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/depthai_nodes/ml/parsers/image_output.py b/depthai_nodes/ml/parsers/image_output.py
@@ -33,7 +33,7 @@ def run(self):
                 raise ValueError(
                     f"Expected 1 output layer, got {len(output_layer_names)}."
                 )
-            output = output.getTensor(output_layer_names[0])
+            output = output.getTensor(output_layer_names[0], dequantize=True)
 
             if len(output.shape) == 4:
                 image = output[0]