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

Web-interface + rest API for classification and regression (https://jeff1evesque.github.io/machine-learning.docs)

Codes and scripts for "Explainable Semantic Space by Grounding Languageto Vision with Cross-Modal Contrastive Learning"

Deep Reinforcement Learning for Multiplayer Online Battle Arena

The code for 'Deep Residual Fourier Transformation for Single Image Deblurring'

Official TensorFlow code for the forthcoming paper

Adversarial Learning for Semi-supervised Semantic Segmentation, BMVC 2018

CV backbones including GhostNet, TinyNet and TNT, developed by Huawei Noah's Ark Lab.

Online Pseudo Label Generation by Hierarchical Cluster Dynamics for Adaptive Person Re-identification

Official Implementation of SWAD (NeurIPS 2021)

Manipulation OpenAI Gym environments to simulate robots at the STARS lab

Lucid Sonic Dreams syncs GAN-generated visuals to music.

TensorFlow (Python) implementation of DeepTCN model for multivariate time series forecasting.

Official implementation of "Accelerating Reinforcement Learning with Learned Skill Priors", Pertsch et al., CoRL 2020

Code of our paper "Contrastive Object-level Pre-training with Spatial Noise Curriculum Learning"

Dilated Convolution with Learnable Spacings PyTorch

Example scripts for the detection of lanes using the ultra fast lane detection model in Tensorflow Lite.

Gym environments used in the paper: "Developmental Reinforcement Learning of Control Policy of a Quadcopter UAV with Thrust Vectoring Rotors"

(CVPR2021) Kaleido-BERT: Vision-Language Pre-training on Fashion Domain

Automatic self-diagnosis program (python required)Automatic self-diagnosis program (python required)