ابدأ مع YOLOv5 🚀 في Docker
مرحبًا بك في دليل البدء السريع لـ Ultralytics YOLOv5 Docker! يقدم هذا البرنامج التعليمي إرشادات خطوة بخطوة لإعداد وتشغيل YOLOv5 داخل حاوية Docker. يتيح لك استخدام Docker تشغيل YOLOv5 في بيئة معزولة ومتسقة، مما يبسط عملية النشر وإدارة التبعيات عبر الأنظمة المختلفة. يعتمد هذا النهج على الحاويات لتجميع التطبيق وتبعياته معًا.
للحصول على طرق إعداد بديلة، ضع في اعتبارك Colab Notebook 
, جهاز افتراضي للتعلم العميق من GCP، أو Amazon AWS أدلة. للحصول على نظرة عامة حول استخدام Docker مع نماذج Ultralytics، انظر إلى دليل البدء السريع لـ Ultralytics Docker.
المتطلبات الأساسية
قبل البدء، تأكد من تثبيت ما يلي:
- Docker: قم بتنزيل وتثبيت Docker من موقع Docker الرسمي. Docker ضروري لإنشاء وإدارة الحاويات.
- برامج تشغيل NVIDIA (مطلوبة لدعم GPU): تأكد من تثبيت برامج تشغيل NVIDIA الإصدار 455.23 أو أعلى. يمكنك تنزيل أحدث برامج التشغيل من موقع NVIDIA على الويب.
- NVIDIA Container Toolkit (مطلوب لدعم GPU): تتيح هذه الأداة لحاويات Docker الوصول إلى وحدات معالجة الرسومات NVIDIA الخاصة بجهازك المضيف. اتبع دليل تثبيت NVIDIA Container Toolkit الرسمي للحصول على إرشادات مفصلة.
إعداد NVIDIA Container Toolkit (لمستخدمي GPU)
أولاً، تحقق من تثبيت برامج تشغيل NVIDIA بشكل صحيح عن طريق تشغيل:
nvidia-smi
يجب أن يعرض هذا الأمر معلومات حول وحدة (وحدات) معالجة الرسوميات (GPU) الخاصة بك وإصدار برنامج التشغيل المثبت.
بعد ذلك، قم بتثبيت NVIDIA Container Toolkit. الأوامر أدناه نموذجية للأنظمة المستندة إلى Debian مثل Ubuntu، ولكن ارجع إلى الدليل الرسمي المرتبط أعلاه للحصول على إرشادات خاصة بالتوزيع الخاص بك:
# Add NVIDIA package repositories (refer to official guide for latest setup)
curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg \
&& curl -s -L https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list \
| sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' \
| sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list
# Update package list and install the toolkit
sudo apt-get update
sudo apt-get install -y nvidia-container-toolkit
# Configure Docker to use the NVIDIA runtime
sudo nvidia-ctk runtime configure --runtime=docker
# Restart Docker service to apply changes
sudo systemctl restart docker
أخيرًا، تحقق من تكوين وقت تشغيل NVIDIA وتوافره لـ Docker:
docker info | grep -i runtime
يجب أن ترى nvidia مدرج كواحد من أوقات التشغيل المتاحة.
الخطوة 1: اسحب صورة Docker الخاصة بـ YOLOv5
توفر Ultralytics صور YOLOv5 الرسمية على Docker Hub. إن latest يشير الوسم إلى أحدث تعديل في المستودع، مما يضمن حصولك دائمًا على أحدث إصدار. اسحب الصورة باستخدام الأمر التالي:
# Define the image name with tag
t=ultralytics/yolov5:latest
# Pull the latest YOLOv5 image from Docker Hub
sudo docker pull $t
يمكنك تصفح جميع الصور المتاحة في مستودع Ultralytics YOLOv5 Docker Hub.
الخطوة الثانية: تشغيل حاوية Docker
بمجرد سحب الصورة، يمكنك تشغيلها كحاوية.
استخدام وحدة المعالجة المركزية (CPU) فقط
لتشغيل مثيل حاوية تفاعلي باستخدام وحدة المعالجة المركزية (CPU) فقط، استخدم -it العلامة. --ipc=host تسمح العلامة بمشاركة مساحة اسم IPC للمضيف، وهو أمر مهم للوصول إلى الذاكرة المشتركة.
# Run an interactive container instance using CPU
sudo docker run -it --ipc=host $t
استخدام وحدة معالجة الرسوميات (GPU)
لتمكين الوصول إلى وحدة معالجة الرسوميات (GPU) داخل الحاوية، استخدم --gpus flag. يتطلب هذا تثبيت NVIDIA Container Toolkit بشكل صحيح.
# Run with access to all available GPUs
sudo docker run -it --ipc=host --gpus all $t
# Run with access to specific GPUs (e.g., GPUs 2 and 3)
sudo docker run -it --ipc=host --gpus '"device=2,3"' $t
راجع مرجع تشغيل Docker لمزيد من التفاصيل حول خيارات الأوامر.
تركيب الدلائل المحلية (Mounting Local Directories)
للعمل مع ملفاتك المحلية (مجموعات البيانات وأوزان النموذج وما إلى ذلك) داخل الحاوية، استخدم -v لتثبيت دليل المضيف في الحاوية:
# Mount /path/on/host (your local machine) to /path/in/container (inside the container)
sudo docker run -it --ipc=host --gpus all -v /path/on/host:/path/in/container $t
استبدل /path/on/host مع المسار الفعلي على جهازك و /path/in/container مع المسار المطلوب داخل حاوية Docker (على سبيل المثال، /usr/src/datasets).
الخطوة الثالثة: استخدام YOLOv5 🚀 داخل حاوية Docker
أنت الآن داخل حاوية Docker الخاصة بتشغيل YOLOv5! من هنا، يمكنك تنفيذ أوامر YOLOv5 القياسية لمهام التعلم الآلي و التعلم العميق المختلفة مثل اكتشاف الكائنات.
# Train a YOLOv5 model on your custom dataset (ensure data is mounted or downloaded)
python train.py --data your_dataset.yaml --weights yolov5s.pt --img 640 # Start training
# Validate the trained model's performance (Precision, Recall, mAP)
python val.py --weights path/to/your/best.pt --data your_dataset.yaml # Validate accuracy
# Run inference on images or videos using a trained model
python detect.py --weights yolov5s.pt --source path/to/your/images_or_videos # Perform detection
# Export the trained model to various formats like ONNX, CoreML, or TFLite for deployment
python export.py --weights yolov5s.pt --include onnx coreml tflite # Export model
استكشف الوثائق للحصول على تفاصيل حول استخدام الأوضاع المختلفة:
تعرف على المزيد حول مقاييس التقييم مثل الدقة، و الاسترجاع، و متوسط الدقة المتوسطة (mAP). فهم تنسيقات التصدير المختلفة مثل ONNX، و CoreML، و TFLite، واستكشف مختلف خيارات نشر النموذج. تذكر إدارة أوزان النموذج الخاصة بك بفعالية.
تهانينا! لقد نجحت في إعداد وتشغيل YOLOv5 داخل حاوية Docker.
