OBB(오리엔티드 바운딩 박스) 데이터 세트 개요

OBB(방향성 경계 상자)로 정밀한 객체 감지 모델을 훈련하려면 철저한 데이터 세트가 필요합니다. 이 가이드에서는 Ultralytics YOLO 모델과 호환되는 다양한 OBB 데이터 세트 형식에 대해 설명하며, 형식 변환을 위한 구조, 응용 및 방법에 대한 인사이트를 제공합니다.

지원되는 OBB 데이터 세트 형식

YOLO OBB 형식

YOLO OBB 형식은 0에서 1 사이로 정규화된 좌표로 네 모서리 점을 기준으로 경계 상자를 지정합니다. 이 형식을 따릅니다:

class_index, x1, y1, x2, y2, x3, y3, x4, y4

내부적으로는 YOLO 에서 손실 및 출력을 처리합니다. xywhr 형식으로 바운딩 박스의 중심점(xy), 너비, 높이 및 회전을 나타냅니다.

의 예 *.txt 클래스 오브젝트가 포함된 위 이미지의 레이블 파일을 생성합니다. 0 는 다음과 같이 보일 수 있습니다:

0 0.780811 0.743961 0.782371 0.74686 0.777691 0.752174 0.776131 0.749758

사용법

이러한 OBB 형식을 사용하여 모델을 학습시키려면 다음과 같이 하세요:

from ultralytics import YOLO

# Create a new YOLOv8n-OBB model from scratch
model = YOLO('yolov8n-obb.yaml')

# Train the model on the DOTAv2 dataset
results = model.train(data='DOTAv1.yaml', epochs=100, imgsz=640)

# Train a new YOLOv8n-OBB model on the DOTAv2 dataset
yolo detect train data=DOTAv1.yaml model=yolov8n.pt epochs=100 imgsz=640

지원되는 데이터 세트

현재 오리엔티드 바운딩 박스가 있는 다음 데이터 집합이 지원됩니다:

• DOTA v2: 항공 이미지에서 물체 감지를 위한 대규모 데이터 세트(DOTA) 버전 2는 항공 관점에서의 감지를 강조하며 170만 개의 인스턴스와 11,268개의 이미지가 포함된 방향성 바운딩 박스를 포함하고 있습니다.

• DOTA8: 전체 DOTA 데이터 세트의 작은 8개 이미지 하위 집합으로, 워크플로 테스트 및 OBB 교육에 대한 지속적 통합(CI) 검사에 적합합니다. ultralytics 리포지토리에 저장합니다.

자체 OBB 데이터 세트 통합하기

방향이 지정된 바운딩 박스가 있는 자체 데이터셋을 도입하려는 경우 위에서 언급한 'YOLO OBB 형식'과 호환되는지 확인하세요. 어노테이션을 이 필수 형식으로 변환하고 해당 YAML 구성 파일에 경로, 클래스 및 클래스 이름을 자세히 설명하세요.

라벨 형식 변환

DOTA 데이터세트 형식을 YOLO OBB 형식으로 변환하기

이 스크립트를 사용하여 DOTA 데이터세트 형식에서 YOLO OBB 형식으로 레이블을 전환할 수 있습니다:

from ultralytics.data.converter import convert_dota_to_yolo_obb

convert_dota_to_yolo_obb('path/to/DOTA')

이 변환 메커니즘은 DOTA 형식의 데이터 세트에 중요한 역할을 하며, Ultralytics YOLO OBB 형식과 일치하도록 보장합니다.

데이터 세트와 모델의 호환성을 검증하고 필요한 형식 규칙을 준수하는 것이 필수적입니다. 적절하게 구조화된 데이터 세트는 방향이 지정된 경계 상자를 사용하여 효율적인 객체 감지 모델을 훈련하는 데 매우 중요합니다.

댓글