diff --git a/depthai_nodes/ml/parsers/image_output.py b/depthai_nodes/ml/parsers/image_output.py index 5b0d88c8..e03e85d6 100644 --- 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]