Skip to content

anarakinson/Drone_Tracking_SSD

Repository files navigation

Object detection and tracking with ssd-mobilenet-320


Создание модели:


Для создания модели используется Tensorflow object-detection API.

Необходимо запустить ноутбук Training_Model.ipynb, добавить снимки для обучения в нужные директории (согласно комментариям в ноутбуке) и сохранить модель для дальнейшего использования. В директорию workspace/models/ можно поместить сохраненные модели, обученные на датасете с сайта kaggle.com, инструкция далее.

Подготовка:


Перед работой нужно скачать архив с сохраненными моделями workspace.zip и распаковать в директорию Tensorflow

Скачать можно по ссылке


Detection and tracking


Для установки среды:

# 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

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published