انتقل إلى المحتوى

الوصف: إتقان Ultralytics YOLO مدرب OBB: أداة متخصصة للتدريب YOLO النماذج باستخدام المربعات المحيطة الموجهة. يتميز بالاستخدام التفصيلي وتهيئة النموذج وعمليات التدريب. الكلمات الرئيسيه: Ultralytics, YOLO مدرب OBB ، صندوق محيط موجه ، تدريب نموذج OBB ، YOLO التدريب النموذجي ، رؤية الكمبيوتر ، التعلم العميق ، التعلم الآلي ، YOLO الكشف عن الكائن ، تهيئة النموذج ، YOLO عملية التدريب


مرجع ل ultralytics/models/yolo/obb/train.py

ملاحظه

هذا الملف متاح في https://github.com/ultralytics/ultralytics/ نقطة / الرئيسية /ultralytics/نماذج/yolo/ obb / القطار .py. إذا اكتشفت مشكلة ، فيرجى المساعدة في إصلاحها من خلال المساهمة في طلب 🛠️ سحب. شكرا لك 🙏!



ultralytics.models.yolo.obb.train.OBBTrainer

قواعد: DetectionTrainer

فئة توسع فئة DetectionTrainer للتدريب بناء على نموذج مربع المحيط الموجه (OBB).

مثل
from ultralytics.models.yolo.obb import OBBTrainer

args = dict(model='yolov8n-obb.pt', data='dota8.yaml', epochs=3)
trainer = OBBTrainer(overrides=args)
trainer.train()
شفرة المصدر في ultralytics/models/yolo/obb/train.py
10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 3233343536373839 404142
class OBBTrainer(yolo.detect.DetectionTrainer):
    """
    A class extending the DetectionTrainer class for training based on an Oriented Bounding Box (OBB) model.

    Example:
        ```python
        from ultralytics.models.yolo.obb import OBBTrainer

        args = dict(model='yolov8n-obb.pt', data='dota8.yaml', epochs=3)
        trainer = OBBTrainer(overrides=args)
        trainer.train()
        ```
    """

    def __init__(self, cfg=DEFAULT_CFG, overrides=None, _callbacks=None):
        """Initialize a OBBTrainer object with given arguments."""
        if overrides is None:
            overrides = {}
        overrides["task"] = "obb"
        super().__init__(cfg, overrides, _callbacks)

    def get_model(self, cfg=None, weights=None, verbose=True):
        """Return OBBModel initialized with specified config and weights."""
        model = OBBModel(cfg, ch=3, nc=self.data["nc"], verbose=verbose and RANK == -1)
        if weights:
            model.load(weights)

        return model

    def get_validator(self):
        """Return an instance of OBBValidator for validation of YOLO model."""
        self.loss_names = "box_loss", "cls_loss", "dfl_loss"
        return yolo.obb.OBBValidator(self.test_loader, save_dir=self.save_dir, args=copy(self.args))

__init__(cfg=DEFAULT_CFG, overrides=None, _callbacks=None)

تهيئة كائن OBBTrainer مع وسيطات معينة.

شفرة المصدر في ultralytics/models/yolo/obb/train.py
def __init__(self, cfg=DEFAULT_CFG, overrides=None, _callbacks=None):
    """Initialize a OBBTrainer object with given arguments."""
    if overrides is None:
        overrides = {}
    overrides["task"] = "obb"
    super().__init__(cfg, overrides, _callbacks)

get_model(cfg=None, weights=None, verbose=True)

قم بإرجاع OBBModel الذي تمت تهيئته باستخدام التكوين والأوزان المحددة.

شفرة المصدر في ultralytics/models/yolo/obb/train.py
def get_model(self, cfg=None, weights=None, verbose=True):
    """Return OBBModel initialized with specified config and weights."""
    model = OBBModel(cfg, ch=3, nc=self.data["nc"], verbose=verbose and RANK == -1)
    if weights:
        model.load(weights)

    return model

get_validator()

إرجاع مثيل OBBValidator للتحقق من صحة YOLO نموذج.

شفرة المصدر في ultralytics/models/yolo/obb/train.py
def get_validator(self):
    """Return an instance of OBBValidator for validation of YOLO model."""
    self.loss_names = "box_loss", "cls_loss", "dfl_loss"
    return yolo.obb.OBBValidator(self.test_loader, save_dir=self.save_dir, args=copy(self.args))





تم إنشاء 2024-01-05, اخر تحديث 2024-01-12
المؤلفون: جلين جوشر (2)