Drone detection using YOLOv5

Last update: Dec 20, 2022

Related tags

Overview

This drone detection system uses YOLOv5 which is a family of object detection architectures and we have trained the model on Drone Dataset.

Overview

Install

Python >= 3.6.0 required with all requirements.txt dependencies installed:

$ git clone https://github.com/tusharsarkar3/Detect_Drone.git
$ pip install -r requirements.txt

Training

The structure of the file system is of great importance here so these images will show you the correct way of organizing it. The main folder named datasets should be on the same level as this repository. The next steps are elaborated in the images:

The two folders with images and labels respectively should be inside the dataset folder.

The images directory should contain the training images and the validation images respectively.

The labels directory should contain the training labels and the validation labels respectively.

Run commands below to reproduce results on Drone Dataset dataset..

$ $ python train.py --img 640 --batch 16 --epochs 15 --data coco128.yaml --weights yolov5s.pt

Check out YOLOv5 for more information.

Inference

$ python detect.py --weights 'path to the best set of weights' --source 0  # webcam       
                                                                        file.jpg  # image 
                                                                        file.mp4  # video
                                                                        path/  # directory
                                                                        path/*.jpg  # glob
                                                                        'https://youtu.be/NUsoVlDFqZg'  # YouTube video
                                                                        'rtsp://example.com/media.mp4'  # RTSP, RTMP, HTTP stream

The results will be stored in a new directory named run which will be on the same level as the root directory.

Check out YOLOv5 for more information.

Drone detection using YOLOv5

Related tags

Overview

Overview

Results:

Developed with ❤️ by Tushar Sarkar

Owner

Tushar Sarkar

Control-Robot-Arm-using-PS4-Controller - A Robotic Arm based on Raspberry Pi and Arduino that controlled by PS4 Controller

An easy way to build PyTorch datasets. Modularly build datasets and automatically cache processed results

Complex Answer Generation For Conversational Search Systems.

Event sourced bank - A wide-and-shallow example using the Python event sourcing library

Official Pytorch implementation of 'GOCor: Bringing Globally Optimized Correspondence Volumes into Your Neural Network' (NeurIPS 2020)

A project for developing transformer-based models for clinical relation extraction

Spherical CNNs

Variational Attention: Propagating Domain-Specific Knowledge for Multi-Domain Learning in Crowd Counting (ICCV, 2021)

Improving Generalization Bounds for VC Classes Using the Hypergeometric Tail Inversion

Use deep learning, genetic programming and other methods to predict stock and market movements

这个开源项目主要是对经典的时间序列预测算法论文进行复现，模型主要参考自GluonTS，框架主要参考自Informer

Code accompanying paper: Meta-Learning to Improve Pre-Training

Tensorflow Implementation for "Pre-trained Deep Convolution Neural Network Model With Attention for Speech Emotion Recognition"

Code for "Graph-Evolving Meta-Learning for Low-Resource Medical Dialogue Generation". [AAAI 2021]

An Evaluation of Generative Adversarial Networks for Collaborative Filtering.

A Kaggle competition: discriminate gender based on handwriting

ReferFormer - Official Implementation of ReferFormer

The official repository for our paper "The Neural Data Router: Adaptive Control Flow in Transformers Improves Systematic Generalization".

Code of the paper "Multi-Task Meta-Learning Modification with Stochastic Approximation".

[AI6122] Text Data Management & Processing