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

Statistical and Algorithmic Investing Strategies for Everyone

Script for getting information in discord

Pairwise Learning for Neural Link Prediction for OGB (PLNLP-OGB)

Proximal Backpropagation - a neural network training algorithm that takes implicit instead of explicit gradient steps

Unofficial implementation of MLP-Mixer: An all-MLP Architecture for Vision

Restricted Boltzmann Machines in Python.

Code repository for the paper "Doubly-Trained Adversarial Data Augmentation for Neural Machine Translation" with instructions to reproduce the results.

Very large and sparse networks appear often in the wild and present unique algorithmic opportunities and challenges for the practitioner

PyTorch implementation of image classification models for CIFAR-10/CIFAR-100/MNIST/FashionMNIST/Kuzushiji-MNIST/ImageNet

Acoustic mosquito detection code with Bayesian Neural Networks

Real life contra a deep learning project built using mediapipe and openc

Expressive Power of Invariant and Equivaraint Graph Neural Networks (ICLR 2021)

Food recognition model using convolutional neural network & computer vision

Robust and Accurate Object Detection via Self-Knowledge Distillation

AutoML library for deep learning

PyTorch implementation of Hierarchical Multi-label Text Classification: An Attention-based Recurrent Network

The Deep Learning with Julia book, using Flux.jl.

Deepfake Scanner by Deepware.

Code of the paper "Performance-Efficiency Trade-offs in Unsupervised Pre-training for Speech Recognition"

SANet: A Slice-Aware Network for Pulmonary Nodule Detection