Add Warmup to classification model

dnum-mi · Jan 22, 2024 · c68902c · c68902c
1 parent fb101c7
commit c68902c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/backend/src/Warmup.jpg b/backend/src/Warmup.jpg
diff --git a/backend/src/model.py b/backend/src/model.py
@@ -28,9 +28,11 @@ def load_model_inference(model_path: str):
         model_path (str): path to model (.pt file)
 
     Returns:
-        Model: loaded model ready for prediction
+        Model: loaded model ready for prediction and Warmud-up
     """
     model = YOLO(model_path)
+    test = Image.open("./Warmup.jpg") 
+    model(test, verbose=False)
     return model