YOLO Vision 2026:

고급 커스텀#

Ultralytics YOLO 커맨드라인 인터페이스와 Python 인터페이스는 모두 기본 엔진 실행기 위에 구축된 고수준 추상화입니다. 이 가이드는 Trainer 엔진에 초점을 맞추며, 특정 요구에 맞게 이를 커스터마이즈하는 방법을 설명합니다.



Watch: Mastering Ultralytics YOLO: Advanced Customization
팁

커스텀 지표, 클래스 가중치 손실, 모델 저장, 백본 프리징, 레이어별 학습률 등 일반적인 트레이너 커스터마이즈의 실용적인 예제는 Customizing Trainer 가이드를 참조하세요.

BaseTrainer#

BaseTrainer 클래스는 다양한 작업에 적응할 수 있는 일반적인 학습 루틴을 제공합니다. 필수 형식을 준수하면서 특정 함수나 작업을 오버라이드하여 커스터마이즈하세요. 예를 들어, 다음 함수들을 오버라이드하여 사용자 정의 모델과 데이터로더를 통합할 수 있습니다:

  • get_model(cfg, weights): 학습할 모델을 빌드합니다.
  • get_dataloader(): 데이터로더를 빌드합니다.

자세한 내용과 소스 코드는 BaseTrainer Reference를 참조하세요.

DetectionTrainer#

Ultralytics YOLO DetectionTrainer을 사용하고 커스터마이즈하는 방법은 다음과 같습니다:

from ultralytics.models.yolo.detect import DetectionTrainer

trainer = DetectionTrainer(overrides={...})
trainer.train()
trained_model = trainer.best  # Get the best model

DetectionTrainer 커스텀하기#

직접 지원되지 않는 커스텀 객체 감지 모델을 학습하려면 기존 get_model 기능을 오버로드하세요:

from ultralytics.models.yolo.detect import DetectionTrainer

class CustomTrainer(DetectionTrainer):
    def get_model(self, cfg=None, weights=None, verbose=True):
        """Loads a custom detection model given configuration and weight files."""

trainer = CustomTrainer(overrides={...})
trainer.train()

loss function을 수정하거나 10 epochs마다 Google Drive에 모델을 업로드하는 callback을 추가하여 트레이너를 더 커스터마이즈하세요. 다음은 예제입니다:

from ultralytics.models.yolo.detect import DetectionTrainer
from ultralytics.nn.tasks import DetectionModel

class MyCustomModel(DetectionModel):
    def init_criterion(self):
        """Initializes the loss function and adds a callback for uploading the model to Google Drive every 10 epochs."""

class CustomTrainer(DetectionTrainer):
    def get_model(self, cfg=None, weights=None, verbose=True):
        """Returns a customized detection model instance configured with specified config and weights."""
        return MyCustomModel(...)

# Callback to upload model weights
def log_model(trainer):
    """Logs the path of the last model weight used by the trainer."""
    last_weight_path = trainer.last
    print(last_weight_path)

trainer = CustomTrainer(overrides={...})
trainer.add_callback("on_train_epoch_end", log_model)  # Adds to existing callbacks
trainer.train()

콜백 트리거 이벤트 및 진입점에 대한 자세한 내용은 Callbacks Guide를 참조하세요.

기타 엔진 컴포넌트#

Validators 및 Predictors과 같은 다른 구성 요소도 비슷하게 커스터마이즈하세요. 자세한 내용은 Validators 및 Predictors 문서를 참조하세요.

커스텀 트레이너와 함께 YOLO 사용하기#

YOLO 모델 클래스는 Trainer 클래스를 위한 고수준 래퍼를 제공합니다. 머신러닝 워크플로에서 더 큰 유연성을 위해 이 아키텍처를 활용할 수 있습니다:

from ultralytics import YOLO
from ultralytics.models.yolo.detect import DetectionTrainer

# Create a custom trainer
class MyCustomTrainer(DetectionTrainer):
    def get_model(self, cfg=None, weights=None, verbose=True):
        """Custom code implementation."""

# Initialize YOLO model
model = YOLO("yolo26n.pt")

# Train with custom trainer
results = model.train(trainer=MyCustomTrainer, data="coco8.yaml", epochs=3)

이 접근 방식을 사용하면 YOLO 인터페이스의 단순성을 유지하면서 특정 요구 사항에 맞게 기본 학습 프로세스를 커스텀할 수 있습니다.

FAQ#

  • 사용자 정의 모델과 데이터로더에 맞게 메서드를 오버라이드하여 특정 작업에 맞게 DetectionTrainer을 커스터마이즈하세요. DetectionTrainer을 상속받는 것으로 시작하고 get_model 같은 메서드를 재정의하여 커스텀 기능을 구현하세요. 다음은 예제입니다:

    from ultralytics.models.yolo.detect import DetectionTrainer

class CustomTrainer(DetectionTrainer):
    def get_model(self, cfg=None, weights=None, verbose=True):
        """Loads a custom detection model given configuration and weight files."""

trainer = CustomTrainer(overrides={...})
trainer.train()
trained_model = trainer.best  # Get the best model

    loss function 변경 또는 callback 추가와 같은 추가 커스터마이즈에 대해서는 Callbacks Guide를 참조하세요.

  • BaseTrainer은 학습 루틴의 기반 역할을 하며, 일반적인 메서드를 오버라이드하여 다양한 작업에 맞게 커스터마이즈할 수 있습니다. 주요 구성 요소는 다음과 같습니다:

    • get_model(cfg, weights): 학습할 모델을 빌드합니다.
    • get_dataloader(): 데이터로더를 빌드합니다.
    • preprocess_batch(): 모델 순전파 전 배치 전처리를 처리합니다.
    • set_model_attributes(): 데이터셋 정보를 기반으로 모델 속성을 설정합니다.
    • get_validator(): 모델 평가를 위한 검증기를 반환합니다.

    커스터마이즈 및 소스 코드에 대한 자세한 내용은 BaseTrainer Reference를 참조하세요.

  • DetectionTrainer에서 학습 과정을 모니터링하고 수정하기 위한 콜백을 추가하세요. 모든 학습 epoch 후에 모델 가중치를 로깅하는 콜백을 추가하는 방법은 다음과 같습니다:

    from ultralytics.models.yolo.detect import DetectionTrainer

# Callback to upload model weights
def log_model(trainer):
    """Logs the path of the last model weight used by the trainer."""
    last_weight_path = trainer.last
    print(last_weight_path)

trainer = DetectionTrainer(overrides={...})
trainer.add_callback("on_train_epoch_end", log_model)  # Adds to existing callbacks
trainer.train()

    콜백 이벤트 및 진입점에 대한 자세한 내용은 Callbacks Guide를 참조하세요.

  • Ultralytics YOLO는 강력한 엔진 실행기 위에 고수준 추상화를 제공하므로 빠른 개발과 커스텀에 이상적입니다. 주요 이점은 다음과 같습니다:

    • 사용 편의성: 커맨드라인 및 Python 인터페이스 모두 복잡한 작업을 단순화합니다.
    • 성능: 실시간 object detection 및 다양한 비전 AI 애플리케이션에 최적화되어 있습니다.
    • 커스터마이즈: 커스텀 모델, loss functions, 데이터로더를 위해 쉽게 확장할 수 있습니다.
    • 모듈화: 전체 파이프라인에 영향을 주지 않고 컴포넌트를 독립적으로 수정할 수 있습니다.
    • 통합: ML 생태계의 인기 있는 프레임워크 및 도구와 원활하게 작동합니다.

    메인 Ultralytics YOLO 페이지를 탐색하여 YOLO의 기능에 대해 자세히 알아보세요.

  • 네, DetectionTrainer은 비표준 모델에 대해 매우 유연하며 커스터마이즈할 수 있습니다. DetectionTrainer을 상속하고 메서드를 오버로드하여 특정 모델의 요구를 지원하세요. 다음은 간단한 예제입니다:

    from ultralytics.models.yolo.detect import DetectionTrainer

class CustomDetectionTrainer(DetectionTrainer):
    def get_model(self, cfg=None, weights=None, verbose=True):
        """Loads a custom detection model."""

trainer = CustomDetectionTrainer(overrides={...})
trainer.train()

    포괄적인 지침과 예제를 보려면 DetectionTrainer Reference를 검토하세요.

기여자
GLglenn-jocher13RAraimbekovm2RIRizwanMunawar2

댓글