| # manga_bubbles_detect |
|
|
| > [English version](README.md) |
|
|
| Модель YOLOv8 для детекции текстовых пузырей на страницах манги. Обнаруживает ограничивающие рамки пузырей (один класс `location-of-bubbles`). |
|
|
| **mAP50 на валидации ≈ 0.977** |
|
|
| Часть проекта [Manga Translate](https://github.com/PSImera/manga_translate) — полный пайплайн перевода манги (детекция пузырей → OCR → inpaint → LLM → отрисовка). |
|
|
|  |
|
|
| ## Использование |
|
|
| ```python |
| from ultralytics import YOLO |
| |
| model = YOLO("bubbles_detect.pt") |
| results = model.predict("page.jpg", conf=0.25, iou=0.5, imgsz=1024) |
| ``` |
|
|
| Или автоматически через приложение [Manga Translate](https://github.com/PSImera/manga_translate) — модель скачивается отсюда при первом запуске. |
|
|
| ## Данные для обучения |
|
|
| Обучена на комбинированном датасете, опубликованном в [PSImera/manga_bubbles_detect](https://huggingface.co/datasets/PSImera/manga_bubbles_detect): |
|
|
| - [manga.v4i (Roboflow)](https://universe.roboflow.com/manga-wtdm0/manga-mvbxx) — 1304/189/103 страниц train/valid/test |
| - 1079 дополнительных страниц из [DLS Manga Translator](https://github.com/ikefir34/DLS_Manga_Translator), откорректированных вручную в CVAT |
|
|
| Скрипты и отчёты обучения: [training/](https://github.com/PSImera/manga_translate/tree/main/training) |
|
|
