feat: add yolo small and xlarge to UltraLytics pipes

BIGmama-technology · Mar 12, 2024 · a265e5d · a265e5d
1 parent 95f3c76
commit a265e5d
Show file tree

Hide file tree

Showing 5 changed files with 10 additions and 0 deletions.
diff --git a/hyko_toolkit/models/computer_vision/ultralytics/image_instance_segmentation/metadata.py b/hyko_toolkit/models/computer_vision/ultralytics/image_instance_segmentation/metadata.py
@@ -15,8 +15,10 @@
 
 class SupportedModels(str, Enum):
     yolov8n = "yolov8_Nano"
+    yolov8s = "yolov8_Small"
     yolov8m = "yolov8_Medium"
     yolov8l = "yolov8_Large"
+    yolov8x = "yolov8_XLarge"
 
 
 @func.set_startup_params

diff --git a/hyko_toolkit/models/computer_vision/ultralytics/image_obb_object_detection/metadata.py b/hyko_toolkit/models/computer_vision/ultralytics/image_obb_object_detection/metadata.py
@@ -15,8 +15,10 @@
 
 class SupportedModels(str, Enum):
     yolov8n = "yolov8_Nano"
+    yolov8s = "yolov8_Small"
     yolov8m = "yolov8_Medium"
     yolov8l = "yolov8_Large"
+    yolov8x = "yolov8_XLarge"
 
 
 @func.set_startup_params

diff --git a/hyko_toolkit/models/computer_vision/ultralytics/image_object_detection/metadata.py b/hyko_toolkit/models/computer_vision/ultralytics/image_object_detection/metadata.py
@@ -15,8 +15,10 @@
 
 class SupportedModels(str, Enum):
     yolov8n = "yolov8_Nano"
+    yolov8s = "yolov8_Small"
     yolov8m = "yolov8_Medium"
     yolov8l = "yolov8_Large"
+    yolov8x = "yolov8_XLarge"
 
 
 @func.set_startup_params

diff --git a/hyko_toolkit/models/computer_vision/ultralytics/video_instance_segmentation/metadata.py b/hyko_toolkit/models/computer_vision/ultralytics/video_instance_segmentation/metadata.py
@@ -15,8 +15,10 @@
 
 class SupportedModels(str, Enum):
     yolov8n = "yolov8_Nano"
+    yolov8s = "yolov8_Small"
     yolov8m = "yolov8_Medium"
     yolov8l = "yolov8_Large"
+    yolov8x = "yolov8_XLarge"
 
 
 @func.set_startup_params

diff --git a/hyko_toolkit/models/computer_vision/ultralytics/video_object_detection/metadata.py b/hyko_toolkit/models/computer_vision/ultralytics/video_object_detection/metadata.py
@@ -15,8 +15,10 @@
 
 class SupportedModels(str, Enum):
     yolov8n = "yolov8_Nano"
+    yolov8s = "yolov8_Small"
     yolov8m = "yolov8_Medium"
     yolov8l = "yolov8_Large"
+    yolov8x = "yolov8_XLarge"
 
 
 @func.set_startup_params