Update YOLOv8n to YOLO11n in args (#16873)
Signed-off-by: UltralyticsAssistant <web@ultralytics.com> Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
This commit is contained in:
parent
60dbee2839
commit
a9d0cf66cb
15 changed files with 53 additions and 53 deletions
|
|
@ -14,7 +14,7 @@ class DetectionPredictor(BasePredictor):
|
|||
from ultralytics.utils import ASSETS
|
||||
from ultralytics.models.yolo.detect import DetectionPredictor
|
||||
|
||||
args = dict(model="yolov8n.pt", source=ASSETS)
|
||||
args = dict(model="yolo11n.pt", source=ASSETS)
|
||||
predictor = DetectionPredictor(overrides=args)
|
||||
predictor.predict_cli()
|
||||
```
|
||||
|
|
|
|||
|
|
@ -24,7 +24,7 @@ class DetectionTrainer(BaseTrainer):
|
|||
```python
|
||||
from ultralytics.models.yolo.detect import DetectionTrainer
|
||||
|
||||
args = dict(model="yolov8n.pt", data="coco8.yaml", epochs=3)
|
||||
args = dict(model="yolo11n.pt", data="coco8.yaml", epochs=3)
|
||||
trainer = DetectionTrainer(overrides=args)
|
||||
trainer.train()
|
||||
```
|
||||
|
|
|
|||
|
|
@ -22,7 +22,7 @@ class DetectionValidator(BaseValidator):
|
|||
```python
|
||||
from ultralytics.models.yolo.detect import DetectionValidator
|
||||
|
||||
args = dict(model="yolov8n.pt", data="coco8.yaml")
|
||||
args = dict(model="yolo11n.pt", data="coco8.yaml")
|
||||
validator = DetectionValidator(args=args)
|
||||
validator()
|
||||
```
|
||||
|
|
|
|||
|
|
@ -11,7 +11,7 @@ from ultralytics.utils import ROOT, yaml_load
|
|||
class YOLO(Model):
|
||||
"""YOLO (You Only Look Once) object detection model."""
|
||||
|
||||
def __init__(self, model="yolov8n.pt", task=None, verbose=False):
|
||||
def __init__(self, model="yolo11n.pt", task=None, verbose=False):
|
||||
"""Initialize YOLO model, switching to YOLOWorld if model filename contains '-world'."""
|
||||
path = Path(model)
|
||||
if "-world" in path.stem and path.suffix in {".pt", ".yaml", ".yml"}: # if YOLOWorld PyTorch model
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue