Update Validator to use model argument (#4480)

This commit is contained in:
Glenn Jocher 2023-08-21 19:21:55 +02:00 committed by GitHub
parent 615ddc9d97
commit b2f279ffdd
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
7 changed files with 15 additions and 14 deletions

View file

@ -22,7 +22,7 @@ class ClassificationValidator(BaseValidator):
args = dict(model='yolov8n-cls.pt', data='imagenet10')
validator = ClassificationValidator(args=args)
validator(model=args['model'])
validator()
```
"""

View file

@ -25,7 +25,7 @@ class DetectionValidator(BaseValidator):
args = dict(model='yolov8n.pt', data='coco8.yaml')
validator = DetectionValidator(args=args)
validator(model=args['model'])
validator()
```
"""

View file

@ -22,7 +22,7 @@ class PoseValidator(DetectionValidator):
args = dict(model='yolov8n-pose.pt', data='coco8-pose.yaml')
validator = PoseValidator(args=args)
validator(model=args['model'])
validator()
```
"""

View file

@ -24,7 +24,7 @@ class SegmentationValidator(DetectionValidator):
args = dict(model='yolov8n-seg.pt', data='coco8-seg.yaml')
validator = SegmentationValidator(args=args)
validator(model=args['model'])
validator()
```
"""