Дана модель була натренована на даних з відкритих джерел, а відповідних даних виявляється не так уже і багато.

Це означає що вона може (і буде) помилятися, не відмічаючи, або відмічаючи не те що потрібно.

Щоб вона якісно працювала потрібно дуже і ДУЖЕ багато відповідного відео/фото для тренувань.

Тому якщо у вас є багато якісного відео з дронів з окупантами надішліть мені на почту sd.sakhro@gmail.com, я буду дуже вдячний! Якщо назбирається достатня кількість якісного відео, це дуже круто покращить результати даної моделі.

Також якщо буде запит, я можу попрацювати більше над моделю для її покращення.

Модель натренована на базі архітектури yolov5n. Також є наявна краща модель на базі yolov5s, але вона важча, і тому не підходить для використання на сайті, або я щось не так зробив :D

В майбутньому планую натренерувати існуючі дані на базі архітектури yolov8, а також по-комбінувати з іншими моделями і архітектурами.

Джерело коду є на github(посилання в хедері). Там є дана модель для роботи з TensorFlow. Якщо потрібні ваги для PyTorch, то напишіть мені на пошту sd.sakhro@gmail.com.