Shahed detect - Yolov5 model

Модель shahed_detect для библиотеки yolov5

Read this in other language: English, Русский, Український

Требования

• python3
• yolov5
• labelImg

Установка

git clone https://github.com/ultralytics/yolov5
cd yolov5
pip3 install -r requirements.txt

Обучение

python3 yolov5/train.py --img 640 --batch -1 --epochs 300 --data ../data/data.yaml --weights yolov5s.pt --cache disk

Пример

python3 yolov5/detect.py --weights models/shahed_detect.pt --source ../doc/raw_1.mp4

Примеры обнаружения