CoolFace
Modelpublic

Aleton/trafficlight

sourceHugging Facemitupdated 4mo agoView on Hugging Face
1likes40downloads
README.md50 linesDownload Raw Back to root
1---2license: mit3library_name: ultralytics4pipeline_tag: object-detection5---6# 🦓 Модель `traficlight.pt` — Обнаружение пешеходных светофоров для мобильных роботов7 8Модель `traficlight.pt` разработана для задачи **обнаружения пешеходных светофоров** с целью распознавания состояния сигнала — **красного** или **зелёного**.  9Она основана на современной архитектуре **YOLOv8** от [Ultralytics](https://github.com/ultralytics/ultralytics) и предназначена для применения в **мобильных роботах**, передвигающихся в городской среде.10 11---12 13## 🎯 Назначение14 15Модель ориентирована на использование в проектах **мобильной робототехники** и системах машинного зрения:16🔴🟢 Определение текущего сигнала светофора — **красный** или **зелёный**17| Зеленый сигнал | Красный сигнал |18|----------|-------------|19| ![input](https://huggingface.co/Aleton/traficlight/resolve/main/green.png) | ![output](https://huggingface.co/Aleton/traficlight/resolve/main/red.jpg) |20 21---22 23## ⚙️ Характеристики24| Параметр                   | Значение                               |25|----------------------------|----------------------------------------|26| 📦 Архитектура             | YOLOv8n                                |27| 📐 Количество слоёв        | 129 (fused: 72)                        |28| 🧮 Параметров              | 3,011,238 (fused: 3,006,038)           |29| 🧠 Градиентов              | 0 (модель в режиме инференса)         |30| ⚙️ FLOPs                   | 8.2 GFLOPs (fused: 8.1 GFLOPs)         |31| 🎯 Количество классов      | 2 (цвета светофора)               |32---33 34## 🚀 Пример использования35 36```python37from ultralytics import YOLO38import cv239 40# Загрузка модели41model = YOLO("traficlight.pt")42 43# Загрузка изображения 44img = cv2.imread("frame.jpg")45 46# Предсказание47results = model(img)48 49# Отображение результата50results[0].show()