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

An atmospheric growth and evolution model based on the EVo degassing model and FastChem 2.0

PyTorch original implementation of Cross-lingual Language Model Pretraining.

Ratatoskr: Worcester Tech's conference scheduling system

TorchCV: A PyTorch-Based Framework for Deep Learning in Computer Vision

PyTorch implementation of PP-LCNet: A Lightweight CPU Convolutional Neural Network

FluxTraining.jl gives you an endlessly extensible training loop for deep learning

All materials of Cassandra Event, Udyam'22

A package for music online and offline rhythmic information analysis including music Beat, downbeat, tempo and meter tracking.

R-Drop: Regularized Dropout for Neural Networks

GULAG: GUessing LAnGuages with neural networks

Supervised forecasting of sequential data in Python.

This is the face keypoint train code of project face-detection-project

A high-performance anchor-free YOLO. Exceeding yolov3~v5 with ONNX, TensorRT, NCNN, and Openvino supported.

Vertical Federated Principal Component Analysis and Its Kernel Extension on Feature-wise Distributed Data based on Pytorch Framework

USAD - UnSupervised Anomaly Detection on multivariate time series

Simple implementation of Mobile-Former on Pytorch

This is the code repository for the paper A hierarchical semantic segmentation framework for computer-vision-based bridge column damage detection

A python code to convert Keras pre-trained weights to Pytorch version

Super Pix Adv - Offical implemention of Robust Superpixel-Guided Attentional Adversarial Attack (CVPR2020)

Large-Scale Unsupervised Object Discovery