نظرة عامة على مجموعات بيانات المربع المحيط الموجه (OBB)

يتطلب تدريب نموذج دقيق للكشف عن الكائنات باستخدام مربعات محيطة موجهة (OBB) مجموعة بيانات شاملة. يشرح هذا الدليل تنسيقات مجموعة بيانات OBB المختلفة المتوافقة مع Ultralytics YOLO النماذج ، التي تقدم رؤى حول هيكلها وتطبيقها وطرق تحويل التنسيق.

تنسيقات مجموعة بيانات OBB المدعومة

YOLO تنسيق OBB

ال YOLO يعين تنسيق OBB المربعات المحيطة بنقاط الزاوية الأربع مع إحداثيات طبيعية بين 0 و 1. يتبع هذا التنسيق:

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

داخليا YOLO يعالج الخسائر والمخرجات في xywhr التنسيق، والذي يمثل نقطة مركز المربع المحيط (xy)، العرض، الارتفاع، والتدوير.

مثال على *.txt ملف التسمية للصورة أعلاه ، والذي يحتوي على كائن من الفئة 0 بتنسيق OBB ، يمكن أن يبدو مثل:

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 ، يؤكد على الكشف من المنظورات الجوية ويحتوي على مربعات محيطة موجهة مع 1.7 مليون مثيل و 11،268 صورة.

• دوتا8مجموعة فرعية صغيرة مكونة من 8 صور من مجموعة بيانات DOTA الكاملة المناسبة لاختبار سير العمل وفحوصات التكامل المستمر (CI) لتدريب OBB في 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.

من الضروري التحقق من توافق مجموعة البيانات مع النموذج الخاص بك والالتزام باصطلاحات التنسيق الضرورية. تعد مجموعات البيانات المنظمة بشكل صحيح أمرا محوريا لتدريب نماذج اكتشاف الكائنات الفعالة باستخدام المربعات المحيطة الموجهة.

التعليقات