Задача
Реализовать демонстрационное приложение, позволяющее детектировать человека в кадре и определять расстояние до него. Приложение должно работать на ноутбуке, задействуя фронтальную камеру ноутбука.
Планируемое использование - демонстрация инвесторам.
Решение
Для детектирования человека использовался фреймворк YOLOv5 с моделью YOLOv5s. Для определения расстояния до человека использовалась модель MiDaS, которая позволяет оценить расстояние по одному изображению.
В задаче было два важных требования:
- Демо-приложение должно было работать на ноутбуке с интегрированной видеокартой. Для этого мы использовали легкие и быстрые модели.
- Расчет расстояния нужно было производить по одному изображению – на момент разработки это было невозможно. Мы доработали модель MiDaS, что позволило выполнить это требования, выдавая корректные оценки с приемлемой точностью.
