Merge pull request #873 from roboflow/fix/confidence-not-applied-in-classification-models

grzegorz-roboflow · web-flow · commit a24d655d2df0 · 2024-12-12T12:10:54.000+01:00
Fix confidence not applied when inferring classification models
diff --git a/inference/core/models/classification_base.py b/inference/core/models/classification_base.py
@@ -162,6 +162,7 @@ def infer(
             disable_preproc_grayscale=disable_preproc_grayscale,
             disable_preproc_static_crop=disable_preproc_static_crop,
             return_image_dims=return_image_dims,
+            **kwargs,
         )
 
     def postprocess(

Original file line number	Diff line number	Diff line change
`@@ -162,6 +162,7 @@ def infer(`
`162`	`162`	`disable_preproc_grayscale=disable_preproc_grayscale,`
`163`	`163`	`disable_preproc_static_crop=disable_preproc_static_crop,`
`164`	`164`	`return_image_dims=return_image_dims,`
	`165`	`+ **kwargs,`
`165`	`166`	`)`
`166`	`167`
`167`	`168`	`def postprocess(`