YOLOV8的主代码是什么？怎么找

最新推荐文章于 2024-09-05 23:02:51 发布

imolly

最新推荐文章于 2024-09-05 23:02:51 发布

阅读量559

点赞数 2

分类专栏： CV101 文章标签： YOLO python 深度学习

本文链接：https://blog.csdn.net/imolly/article/details/134537972

版权

CV101 专栏收录该内容

3 篇文章 0 订阅

订阅专栏

YOLOv8官方公开代码：https://github.com/ultralytics/ultralytics/tree/main/ultralytics

我自己的寻找过程

ultralytics/models/yolo/model.py

```
# Ultralytics YOLO 🚀, AGPL-3.0 license

from ultralytics.engine.model import Model
from ultralytics.models import yolo  # noqa
"""
下面的导入是重点！
它导入了各种应用的model，所以【model们来自的 ultralytics.nn.tasks 就是YOLOv8的主代码】
"""
from ultralytics.nn.tasks import ClassificationModel, DetectionModel, PoseModel, SegmentationModel


class YOLO(Model):
    """YOLO (You Only Look Once) object detection model."""

    @property
    def task_map(self):
        """Map head to model, trainer, validator, and predictor classes."""
        return {
            'classify': {
                'model': ClassificationModel,
                'trainer': yolo.classify.ClassificationTrainer,
                'validator': yolo.classify.ClassificationValidator,
                'predictor': yolo.classify.ClassificationPredictor, },
            'detect': {
                'model': DetectionModel,
                'trainer': yolo.detect.DetectionTrainer,
                'validator': yolo.detect.DetectionValidator,
                'predictor': yolo.detect.DetectionPredictor, },
            'segment': {
                'model': SegmentationModel,
                'trainer': yolo.segment.SegmentationTrainer,
                'validator': yolo.segment.SegmentationValidator,
                'predictor': yolo.segment.SegmentationPredictor, },
            'pose': {
                'model': PoseModel,
                'trainer': yolo.pose.PoseTrainer,
                'validator': yolo.pose.PoseValidator,
                'predictor': yolo.pose.PosePredictor, }, }
```