Данный проект представляет собой телеграмм бота, использующего методы искусственного интеллекта для автоматического подсчёта труб на фотографиях. Бот способен распознавать два типа труб: круглые и квадратные/прямоугольные.
Проект состоит из двух основных папок:
- Artificial intelligence methods:
- В этой папке находятся скрипты, отвечающие за подсчёт труб и улучшение качества фотографий. Подробности о работе каждого скрипта можно найти в файле README, который находится внутри этой папки.
- telegram interface:
- Здесь расположен код для запуска телеграмм бота и дополнительные конструкции, необходимые для его функционирования.
Все необходимые библиотеки можно найти в соответствующих папках, а также инструкции по установке в файлах README.
Ниже представлены примеры изображений, на которых бот успешно распознаёт трубы:
-
Подсчёт прямоугольных труб:
-
Подсчёт круглых труб:
- Модель YOLOv8: Используется для распознавания объектов на изображениях.
- Сегментация: Применяется для выделения контуров труб, что позволяет повысить точность подсчёта.
В проекте реализовано распознавание двух категорий труб:
- Круглые
- Квадратные/Прямоугольные
-
Запустите телеграмм бота, следуя инструкциям в папке telegram interface .
-
Выберете категорию трубы для распознавания.
-
Отправьте фотографию с трубами боту.
-
Получите результат подсчёта труб на изображении.