Fix yolov8_classification params

roboflow · Jan 17, 2025 · be941dc · be941dc
1 parent c0cf066
commit be941dc
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/inference/core/models/classification_base.py b/inference/core/models/classification_base.py
@@ -35,6 +35,9 @@ class ClassificationBaseOnnxRoboflowInferenceModel(OnnxRoboflowInferenceModel):
 
     task_type = "classification"
 
+    preprocess_means = [0.5, 0.5, 0.5]
+    preprocess_stds = [0.5, 0.5, 0.5]
+
     def __init__(self, *args, **kwargs):
         """Initialize the model, setting whether it is multiclass or not."""
         super().__init__(*args, **kwargs)