Aleton/trafficlight

🦓 Модель traficlight.pt — Обнаружение пешеходных светофоров для мобильных роботов

Модель traficlight.pt разработана для задачи обнаружения пешеходных светофоров с целью распознавания состояния сигнала — красного или зелёного. Она основана на современной архитектуре YOLOv8 от Ultralytics и предназначена для применения в мобильных роботах, передвигающихся в городской среде.

🎯 Назначение

Модель ориентирована на использование в проектах мобильной робототехники и системах машинного зрения: 🔴🟢 Определение текущего сигнала светофора — красный или зелёный | Зеленый сигнал | Красный сигнал | |----------|-------------| | | |

⚙️ Характеристики

🚀 Пример использования

from ultralytics import YOLO
import cv2

# Загрузка модели
model = YOLO("traficlight.pt")

# Загрузка изображения 
img = cv2.imread("frame.jpg")

# Предсказание
results = model(img)

# Отображение результата
results[0].show()