구성

YOLO settings and hyperparameters play a critical role in the model's performance, speed, and accuracy. These settings and hyperparameters can affect the model's behavior at various stages of the model development process, including training, validation, and prediction.

Watch: Mastering Ultralytics YOLO: Configuration

Ultralytics 명령은 다음 구문을 사용합니다:

yolo TASK MODE ARGS

from ultralytics import YOLO

# Load a YOLO11 model from a pre-trained weights file
model = YOLO("yolo11n.pt")

# Run MODE mode using the custom arguments ARGS (guess TASK)
model.MODE(ARGS)

Where:

• TASK (선택 사항)은 (감지, 세그먼트, 분류, 포즈, obb)
• MODE (필수)는 (기차, val, 예측, 내보내기, 트랙, 벤치마크)
• ARGS (선택 사항)은 다음과 같습니다. arg=value 같은 imgsz=640 기본값을 재정의할 수 있습니다.

기본값 ARG 값은 이 페이지에서 cfg/defaults.yaml 파일.

작업

YOLO 모델은 감지, 세분화, 분류, 포즈 등 다양한 작업에 사용할 수 있습니다. 이러한 작업은 생성하는 결과물의 유형과 해결하고자 하는 특정 문제가 다릅니다.

• 감지: 감지: 이미지 또는 동영상에서 관심 있는 물체 또는 영역을 식별하고 위치를 파악합니다.
• 세그먼트: 이미지 또는 동영상을 다른 객체 또는 클래스에 해당하는 영역 또는 픽셀로 나누는 데 사용됩니다.
• 분류: 입력 이미지의 클래스 레이블을 예측합니다.
• 포즈: 포즈: 이미지 또는 동영상에서 객체를 식별하고 키포인트를 추정하는 데 사용됩니다.
• OBB: 위성 또는 의료 이미지에 적합한 방향성(즉, 회전된) 바운딩 박스입니다.

작업 가이드

모드

YOLO 모델은 해결하려는 특정 문제에 따라 다양한 모드로 사용할 수 있습니다. 이러한 모드에는 다음이 포함됩니다:

• Train: For training a YOLO11 model on a custom dataset.
• Val: For validating a YOLO11 model after it has been trained.
• Predict: For making predictions using a trained YOLO11 model on new images or videos.
• Export: For exporting a YOLO11 model to a format that can be used for deployment.
• Track: For tracking objects in real-time using a YOLO11 model.
• Benchmark: For benchmarking YOLO11 exports (ONNX, TensorRT, etc.) speed and accuracy.

모드 가이드

열차 설정

The training settings for YOLO models encompass various hyperparameters and configurations used during the training process. These settings influence the model's performance, speed, and accuracy. Key training settings include batch size, learning rate, momentum, and weight decay. Additionally, the choice of optimizer, loss function, and training dataset composition can impact the training process. Careful tuning and experimentation with these settings are crucial for optimizing performance.

기차 가이드

예측 설정

YOLO 모델의 예측 설정에는 새로운 데이터를 추론하는 동안 모델의 성능, 속도 및 정확도에 영향을 미치는 다양한 하이퍼파라미터와 구성이 포함되어 있습니다. 특정 작업에 대한 최적의 성능을 얻으려면 이러한 설정을 신중하게 조정하고 실험해 보는 것이 필수적입니다. 주요 설정에는 신뢰도 임계값, NMS(비최대 억제) 임계값 및 고려되는 클래스 수가 포함됩니다. 예측 프로세스에 영향을 미치는 추가 요소로는 입력 데이터 크기 및 형식, 마스크 또는 상자당 여러 개의 레이블과 같은 보조 기능의 존재 여부, 모델이 사용되는 특정 작업 등이 있습니다.

추론 인수:

시각화 인수:

예측 가이드

유효성 검사 설정

YOLO 모델의 VAL(유효성 검사) 설정에는 유효성 검사 데이터 세트에서 모델의 성능을 평가하는 데 사용되는 다양한 하이퍼파라미터 및 구성이 포함됩니다. 이러한 설정은 모델의 성능, 속도 및 정확도에 영향을 미칩니다. 일반적인 YOLO 유효성 검사 설정에는 배치 크기, 학습 중 유효성 검사 빈도 및 성능 평가 메트릭이 포함됩니다. 유효성 검사 프로세스에 영향을 미치는 다른 요인으로는 유효성 검사 데이터 세트의 크기와 구성, 모델이 사용되는 특정 작업 등이 있습니다.

Careful tuning and experimentation with these settings are crucial to ensure optimal performance on the validation dataset and detect and prevent overfitting.

Val 가이드

내보내기 설정

YOLO 모델의 내보내기 설정에는 다양한 환경이나 플랫폼에서 사용할 수 있도록 모델을 저장하거나 내보내는 것과 관련된 구성 및 옵션이 포함됩니다. 이러한 설정은 모델의 성능, 크기 및 다양한 시스템과의 호환성에 영향을 줄 수 있습니다. 주요 내보내기 설정에는 내보내는 모델 파일 형식(예: ONNX, TensorFlow SavedModel ), 대상 장치(예: CPU, GPU), 마스크 또는 상자당 여러 개의 레이블과 같은 추가 기능이 포함됩니다. 내보내기 프로세스는 모델의 특정 작업과 대상 환경 또는 플랫폼의 요구 사항 또는 제약 조건에 따라 영향을 받을 수도 있습니다.

내보낸 모델이 의도한 사용 사례에 최적화되고 대상 환경에서 효과적으로 작동하도록 하려면 이러한 설정을 신중하게 구성하는 것이 중요합니다.

내보내기 가이드

증강 설정

Augmentation techniques are essential for improving the robustness and performance of YOLO models by introducing variability into the training data, helping the model generalize better to unseen data. The following table outlines the purpose and effect of each augmentation argument:

이러한 설정은 데이터 세트와 작업의 특정 요구 사항을 충족하도록 조정할 수 있습니다. 다양한 값으로 실험해 보면 최상의 모델 성능을 이끌어내는 최적의 증강 전략을 찾는 데 도움이 될 수 있습니다.

로깅, 체크포인트 및 플로팅 설정

로깅, 체크포인트, 플로팅 및 파일 관리는 YOLO 모델을 학습시킬 때 중요한 고려 사항입니다.

• 로깅: 모델의 진행 상황을 추적하고 발생할 수 있는 문제를 진단하기 위해 훈련 중에 다양한 메트릭과 통계를 기록하는 것이 도움이 되는 경우가 많습니다. 이 작업은 텐서보드와 같은 로깅 라이브러리를 사용하거나 로그 메시지를 파일에 기록하여 수행할 수 있습니다.
• 체크포인트: 훈련 중 일정한 간격으로 모델의 체크포인트를 저장하는 것이 좋습니다. 이렇게 하면 학습 프로세스가 중단되거나 다른 학습 구성을 실험하려는 경우 이전 지점부터 학습을 재개할 수 있습니다.
• 플로팅: 모델의 성능과 학습 진행 상황을 시각화하면 모델의 작동 방식을 이해하고 잠재적인 문제를 파악하는 데 도움이 될 수 있습니다. 이 작업은 matplotlib와 같은 플로팅 라이브러리를 사용하거나 TensorBoard와 같은 로깅 라이브러리를 사용하여 플롯을 생성하여 수행할 수 있습니다.
• 파일 관리: 모델 체크포인트, 로그 파일, 플롯 등 훈련 과정에서 생성되는 다양한 파일을 관리하는 것은 어려울 수 있습니다. 이러한 파일을 추적하고 필요에 따라 쉽게 액세스하고 분석할 수 있도록 명확하고 체계적인 파일 구조를 갖추는 것이 중요합니다.

효과적인 로깅, 체크포인트, 플로팅 및 파일 관리를 통해 모델의 진행 상황을 추적하고 훈련 프로세스를 더 쉽게 디버깅하고 최적화할 수 있습니다.

자주 묻는 질문

교육 중에 YOLO 모델의 성능을 향상시키려면 어떻게 해야 하나요?

Improving YOLO model performance involves tuning hyperparameters like batch size, learning rate, momentum, and weight decay. Adjusting augmentation settings, selecting the right optimizer, and employing techniques like early stopping or mixed precision can also help. For detailed guidance on training settings, refer to the Train Guide.

YOLO 모델 정확도를 위해 고려해야 할 주요 하이퍼파라미터는 무엇인가요?

YOLO 모델 정확도에 영향을 미치는 주요 하이퍼파라미터는 다음과 같습니다:

• 배치 크기(batch): 배치 크기가 클수록 훈련이 안정화될 수 있지만 더 많은 메모리가 필요할 수 있습니다.
• 학습률 (lr0): 가중치 업데이트의 단계 크기를 제어합니다. 속도가 작을수록 미세 조정이 가능하지만 수렴 속도가 느려집니다.
• 모멘텀 (momentum): 그래디언트 벡터를 올바른 방향으로 가속하여 진동을 완화합니다.
• 이미지 크기(imgsz): 이미지 크기가 클수록 정확도는 향상되지만 계산 부하가 증가할 수 있습니다.

데이터 세트와 하드웨어 기능에 따라 이 값을 조정하세요. 훈련 설정 섹션에서 자세히 알아보세요.

YOLO 모델 학습을 위한 학습률은 어떻게 설정하나요?

학습 속도(lr0)는 최적화에 매우 중요합니다. 일반적인 시작점은 0.01 또는 0.001 를 설정합니다. 학습 메트릭을 모니터링하고 필요한 경우 조정하는 것이 중요합니다. 코사인 학습률 스케줄러 사용(cos_lr) 또는 워밍업 기술(warmup_epochs, warmup_momentum)를 사용하여 훈련 중에 속도를 동적으로 수정할 수 있습니다. 자세한 내용은 기차 가이드.

YOLO 모델의 기본 추론 설정은 무엇인가요?

기본 추론 설정에는 다음이 포함됩니다:

• 신뢰도 임계값(conf=0.25): 탐지에 대한 최소 신뢰도.
• IoU 임계값(iou=0.7): 비 최대 억제(NMS)의 경우.
• 이미지 크기(imgsz=640): 추론하기 전에 입력 이미지 크기를 조정합니다.
• 장치 (device=None): Selects CPU or GPU for inference. For a comprehensive overview, visit the 예측 설정 섹션과 예측 가이드.

혼합 정밀 교육을 YOLO 모델과 함께 사용해야 하는 이유는 무엇인가요?

혼합 정밀 교육, 다음을 통해 활성화 amp=True를 사용하면 메모리 사용량을 줄이고 FP16과 FP32의 장점을 모두 활용하여 훈련 속도를 높일 수 있습니다. 이는 기본적으로 혼합 정밀도를 지원하는 최신 GPU에 유용하며, 더 많은 모델을 메모리에 넣을 수 있고 정확도의 큰 손실 없이 더 빠른 계산을 가능하게 합니다. 이에 대한 자세한 내용은 기차 가이드.

📅 Created 11 months ago ✏️ Updated 17 days ago

댓글