YOLO26s Serial Number Bounding Box
детектор серийных номеров для OCR
Это fine-tune YOLO26s (small) от nuven-mira для поиска строки серийного номера на промышленных фото счётчиков. Модель выдаёт один bounding box на номер, который можно вырезать и передать в OCR. Обучена на 8695 размеченных изображениях, один класс serial_number.
🔮 Ключевые особенности
- 📷 Детекция одного класса
serial_number: один bounding box на серийный номер. - 🔥 База YOLO26s (Ultralytics, 260 слоёв), 9,948,638 параметров и 22.7 GFLOPs.
- 💻 Вход 640 px; на RTX 3050 batch 1 — 10.7 мс/изображение, около 93 FPS.
- ✅ На тесте mAP50 0.9748 и mAP50-95 0.7980; рекомендуемый conf — 0.39.
- 📚 Обучение на 8695 фото электросчётчиков с 8696 bbox; датасет CC BY 4.0.
💻 Пример
from ultralytics import YOLO
model = YOLO("yolo26s_serial-small-10-11-26.pt")
res = model.predict("meter_photo.jpg", imgsz=640, conf=0.39)[0]
for box in res.boxes:
x1, y1, x2, y2 = box.xyxy[0].tolist()
conf = float(box.conf[0])
Подробнее
На 870 тестовых изображениях: mAP50 0.9748, mAP50-95 0.7980; при conf 0.39 — P 0.9702, R 0.9724, F1 0.9713.Обучение: 100 эпох с early stopping, batch 16, imgsz 640; около 4.3 ч на одной RTX 3050 8 GB.
Ограничения: грязное или отражающее стекло — самые сложные случаи; модель обучена на фото передних панелей счётчиков, для других товаров может потребоваться fine-tune. Чтение цифр — отдельный OCR-шаг.
Лицензия CC BY 4.0.








Комментарии