|
|
@ -3,7 +3,6 @@ from ultralytics import YOLO
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class UltralyticsModel(BaseMultiModalModel):
|
|
|
|
class UltralyticsModel(BaseMultiModalModel):
|
|
|
|
def __init__(self, model_name: str, *args, **kwargs):
|
|
|
|
|
|
|
|
"""
|
|
|
|
"""
|
|
|
|
Initializes an instance of the Ultralytics model.
|
|
|
|
Initializes an instance of the Ultralytics model.
|
|
|
|
|
|
|
|
|
|
|
@ -12,6 +11,7 @@ class UltralyticsModel(BaseMultiModalModel):
|
|
|
|
*args: Variable length argument list.
|
|
|
|
*args: Variable length argument list.
|
|
|
|
**kwargs: Arbitrary keyword arguments.
|
|
|
|
**kwargs: Arbitrary keyword arguments.
|
|
|
|
"""
|
|
|
|
"""
|
|
|
|
|
|
|
|
def __init__(self, model_name: str, *args, **kwargs):
|
|
|
|
super().__init__(*args, **kwargs)
|
|
|
|
super().__init__(*args, **kwargs)
|
|
|
|
self.model_name = model_name
|
|
|
|
self.model_name = model_name
|
|
|
|
|
|
|
|
|
|
|
|