Views
No views yet
hybrid_active_learning_newspaper_ocr.md в
исходном репозитории на GitHub (ссылка появится после публикации).| Версия | Train/Val страниц | Классов | mAP50 (overall) | Что изменилось |
|---|---|---|---|---|
v1_retry | 40/10 | 11 | 0.614 | Первая модель, эталонная разметка 50 страниц |
v2 | 51/14 | 11 | 0.890 | + 15 доразмеченных сложных страниц (фоторепортажи) |
v3 | 51/14 | 9 | 0.958 | Убраны редкие шумные классы caption, continuation_marker |
v3 — самая точная и стабильная версия.0: headline — заголовок статьи
1: text_column — колонка текста
2: photo — фотография/иллюстрация
3: author — автор статьи
4: page_number — номер страницы
5: masthead — шапка газеты
6: footer — футер страницы
7: teaser — анонс другой статьи
8: junk — мусор, декоративные элементы| Класс | mAP50 |
|---|---|
| headline | 0.952 |
| text_column | 0.984 |
| photo | 0.964 |
| author | 0.904 |
| masthead | 0.990 |
| footer | 0.995 |
| teaser | 0.913 |
1from ultralytics import YOLO
2
3model = YOLO("v3/best.pt")
4results = model("page.png")
5
6for box in results[0].boxes:
7 cls_name = model.names[int(box.cls)]
8 conf = float(box.conf)
9 print(f"{cls_name}: {conf:.3f}")