hotfix: yolov8 warmup

sieve-community · Aug 27, 2024 · b95f635 · b95f635
1 parent 93d80a5
commit b95f635
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/object_detection/yolov8/test-image.jpg b/object_detection/yolov8/test-image.jpg
diff --git a/object_detection/yolov8/yolov8_model.py b/object_detection/yolov8/yolov8_model.py
@@ -47,6 +47,19 @@ def __setup__(self):
         self.current_world_classes = None
         self.current_world_fast_classes = None
 
+        test_file = sieve.File("test-image.jpg")
+
+        print("Warming up inference...")
+
+        self.__predict__(test_file)
+        self.__predict__(test_file, models="yolov8n")
+        self.__predict__(test_file, models="yolov8l-face")
+        self.__predict__(test_file, models="yolov8n-face")
+        self.__predict__(test_file, models="yolov8s-world")
+        self.__predict__(test_file, models="yolov8l-world")
+
+        print("Inference warmed up.")
+
     def __predict__(
             self,
             file: sieve.File,