Для создания модели используется Tensorflow object-detection API.
Необходимо запустить ноутбук Training_Model.ipynb, добавить снимки для обучения в нужные директории (согласно комментариям в ноутбуке) и сохранить модель для дальнейшего использования. В директорию workspace/models/ можно поместить сохраненные модели, обученные на датасете с сайта kaggle.com, инструкция далее.
Перед работой нужно скачать архив с сохраненными моделями workspace.zip и распаковать в директорию Tensorflow
Скачать можно по ссылке
Для установки среды:
# windows
python -m venv venv
#linux
python -m virtualenv venv
Установка необходимых библиотек:
pip install -r requirements.txt
python setup.py
Проверка установки библиотек для детекции:
python verification.py
Детекция и трекинг производятся с веб-камеры
Детекция:
python detect.py
Трекинг:
python object_trecking.py