Поиск неисправностей ЖД пути по видео
Проект выполнен в рамках хакатона "Стартап депо".
г.Хабаровск 11-13 декабря 2020 года.
Наша команда заняла первое место, и получила приз 150 000 рублей!
Необходимо проанализировать видеофайл со съемкой ЖД-пути, и найти на нем неисправности. Подробное задание
- Поиск стыков методом поиска прямых линий в кадре. Ссылка на ноутбук
- Строим маршрут рельса, и ищем типичные (для стыка) отклонения. Ссылка на ноутбук
- Детекция объектов (стык, болт) на фотографии со стыком. Ссылка на ноутбуки
- Разработка GUI приложения. Ссылка на исходники
Блоконоты jupyter notebook из разделов 1, 2 и 3 запускались в среде Anacondа
Установка зависимостей, если это требуется, проиходит прямо в них.
Запуск финальной программы
- Скачать исходные данные и обученную модель из этого файла, положить их в директорию
data
в корне проекта - Установить программу
Tesseract OCR
отсюда - В файле
4. GUI application\config.py
указать путь до exe-файла tesseract
pytesseract.pytesseract.tesseract_cmd = r"C:\ПУТЬ\ДО\КАТАЛОГА\УСТАНОВКИ\tesseract.exe"
- Установить необходимые библиотеки python
pip install -r "4. GUI application\requirements.txt"
pip install torch==1.7.1+cpu torchvision==0.8.2+cpu torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
git clone https://github.com/DGMaxime/detectron2-windows.git
pip install -e detectron2-windows
- строка установки pytorch определяется на сайте pytorch
- detectron2 устанавливается из неофициально форка (официальная версия не поддерживает платформу Windows)
- Запустить программу
python "4. GUI application\defect_detector.py"
- Антон Фролов
- Кирилл Шоберг
- Юрий Сухобок
- Роман Заболотин