كيفية عرض نتائج استدلال YOLO في طرفية VSCode#

صورة من موقع libsixel.

الدافع#

عند الاتصال بجهاز بعيد عبر SSH، لا يكون تصور نتائج الصور ممكنًا في العادة أو يتطلب نقل البيانات إلى جهاز محلي به واجهة رسومية GUI. يمكن للمحطة الطرفية المدمجة في VSCode عرض الصور مباشرة، لذا يمكنك فحص نتائج التنبؤ تمامًا حيث تشغل الاستدلال، دون الحاجة إلى نسخ الملفات مرة أخرى إلى حاسوبك المحمول. يوضح هذا الدليل خطوات تفعيل الإعداد، وربط الكود، ويجيب عن الأسئلة الشائعة.

البروتوكولات المتوافقة مع VSCode لعرض الصور باستخدام المحطة الطرفية المدمجة هي sixel و iTerm. سيوضح هذا الدليل استخدام بروتوكول sixel.

العملية#

1. أولاً، يجب عليك تفعيل الإعدادين terminal.integrated.enableImages و terminal.integrated.gpuAcceleration في VSCode.

   "terminal.integrated.gpuAcceleration": "auto" # "auto" is default, can also use "on"
   "terminal.integrated.enableImages": true

   

2. قم بتثبيت مكتبة python-sixel في بيئتك الافتراضية. هذه نسخة متفرعة (fork) من مكتبة PySixel التي لم تعد صاشحة للصيانة.

   pip install sixel

3. قم بتحميل النموذج وتنفيذ الاستدلال، ثم ارسم النتائج وخزنها في متغير. راجع المزيد حول وسائط الاستدلال والتعامل مع النتائج على صفحة وضع التنبؤ.

   from ultralytics import YOLO
   
   # Load a model
   model = YOLO("yolo26n.pt")
   
   # Run inference on an image
   results = model.predict(source="ultralytics/assets/bus.jpg")
   
   # Plot inference results
   plot = results[0].plot()  # (1)!

   1. راجع معلمات طريقة الرسم لمعرفة الوسائط الممكنة للاستخدام.

4. الآن، استخدم OpenCV لتحويل بيانات np.ndarray إلى bytes. ثم استخدم io.BytesIO لإنشاء كائن "يشبه الملف (file-like)".

   import io
   
   import cv2
   
   # Results image as bytes
   im_bytes = cv2.imencode(
       ".png",  # (1)!
       plot,
   )[1].tobytes()  # (2)!
   
   # Image bytes as a file-like object
   mem_file = io.BytesIO(im_bytes)

   1. من الممكن استخدام امتدادات صور أخرى أيضاً.
   2. الكائن الوحيد المطلوب هو الكائن الذي يتم إرجاعه عند الفهرس 1.

5. أنشئ مثيلاً لـ SixelWriter، ثم استخدم طريقة .draw() لرسم الصورة في المحطة الطرفية.

   from sixel import SixelWriter
   
   # Create sixel writer object
   w = SixelWriter()
   
   # Draw the sixel image in the terminal
   w.draw(mem_file)

مثال على نتائج الاستدلال#

مثال كامل للكود#

import io

import cv2
from sixel import SixelWriter

from ultralytics import YOLO

# Load a model
model = YOLO("yolo26n.pt")

# Run inference on an image
results = model.predict(source="ultralytics/assets/bus.jpg")

# Plot inference results
plot = results[0].plot()  # (3)!

# Results image as bytes
im_bytes = cv2.imencode(
    ".png",  # (1)!
    plot,
)[1].tobytes()  # (2)!

mem_file = io.BytesIO(im_bytes)
w = SixelWriter()
w.draw(mem_file)

1. من الممكن استخدام امتدادات صور أخرى أيضاً.
2. الكائن الوحيد المطلوب هو الكائن الذي يتم إرجاعه عند الفهرس 1.
3. راجع معلمات طريقة الرسم لمعرفة الوسائط الممكنة للاستخدام.

الأسئلة الشائعة#