CheckDocAI

Телеграм-бот с ИИ модулем для контроля качества оформления документов

Задача

Создать сервис, обеспечивающий проверку корректности и полноты оформления Актов проверки работоспособности системы автоматической пожарной сигнализации. Сервис должен принимать PDF-файл с отсканированными актами и возвращать информацию о найденных дефектах. Критерий проверки: на каждом акте должны быть две печати и две подписи.

Команда проекта

  • Михаил Васильев
  • старший специалист по машинному обучению
  • Маквес
  • @LaHundo
  • тим-лид и технический эксперт
  • Юлия Сухинина
  • руководитель по развитию
  • Гольфстрим
  • @Sukhinina_Julia
  • разметка данных, аугментации, обучение модели

Технологии

yolo, albumentations, aiogram, cvat

Решение

(репозиторий)

Собрали собственный датасет с отсканированными Актами проверки. Произвели разметку датасета с использованием сервиса CVAT. Для каждого акта мы размечали проставленные печати, подписи, даты, а также номер акта. Произвели аугментацию датасета с использованием библиотеки albumentations. Обучили нейросеть YOLO v8.

Для взаимодействия с пользователем реализовали телеграм-бота, принимающего на вход PDF-файл с отсканированными актами и возвращающего результаты проверки.

Решение развёрнуто и успешно используется АО «Гольфстрим охранные системы».

Доклад

По результатом проекта мы выступили с докладом на митапе Moscow Python Meetup № 95. Рассказали об идее проекта и сложностях, с которыми столкнулись в процессе разработки.

Презентация