diff --git a/object_detection/yolov8/test-image.jpg b/object_detection/yolov8/test-image.jpg new file mode 100644 index 0000000..7fb175b Binary files /dev/null and b/object_detection/yolov8/test-image.jpg differ diff --git a/object_detection/yolov8/yolov8_model.py b/object_detection/yolov8/yolov8_model.py index 03c3187..ee72270 100644 --- 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,