tracktor

tracktor это обертка над моделью YOLOv8 для определение объектов из датасета COCO 2017 на видео и генерации видео с найденным вырезанными объектами.

Установка

Необходимо склонировать репозиторий:

> git clone https://github.com/samedit66/tracktor.git
> cd tracktor

Установка в виртуальной среде

По желанию можно создать виртуальную среду, чтобы не загрязнять глобальное пространство пакетов:

> python -m venv venv

Активация виртуальной среды

Windows (cmd):

> venv\Scripts\activate.bat

Windows (powershell):

> venv\Scripts\Activate.ps1

Linux и MacOS:

> source venv/bin/activate

Установка зависимостей:

> pip install -r requirements.txt

Запуск

Требуется соединение с интернетом в случае, если в папке нет модели YOLO. Помимо этого Ultralytics может поставить дополнительный пакет lapx, поэтому при первом запуске желательно иметь интернет соединение. Для запуска необходимо задать индекс искомого класса через флаг --class и передать путь к видео (класс 15 - коты):

> python .\tracktor.py --class 15 "путь к видео"

В результате будет сгенерирован файл output.avi, содержащий трекинг и обрезку объектов. Через флаг --yolo имеется возможность задать модель для определения объектов (по умолчанию используется yolov8n.pt):

> python .\tracktor.py --class 15 --yolo my_custom_yolo.pt video.mp4