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

Unofficial Implementation of MLP-Mixer in TensorFlow

The Incredible PyTorch: a curated list of tutorials, papers, projects, communities and more relating to PyTorch.

HyperPose is a library for building high-performance custom pose estimation applications.

Implementation of a protein autoregressive language model, but with autoregressive infilling objective (editing subsequences capability)

Pytorch implementation of NeurIPS 2021 paper: Geometry Processing with Neural Fields.

PyTorch Implementation for Fracture Detection in Wrist Bone X-ray Images

Brain Tumor Detection with Tensorflow Neural Networks.

Repo for "Benchmarking Robustness of 3D Point Cloud Recognition against Common Corruptions" https://arxiv.org/abs/2201.12296

Self Driving RC Car Code

A privacy-focused, intelligent security camera system.

implementation of paper - You Only Learn One Representation: Unified Network for Multiple Tasks

Adabelief-Optimizer - Repository for NeurIPS 2020 Spotlight "AdaBelief Optimizer: Adapting stepsizes by the belief in observed gradients"

In this project we investigate the performance of the SetCon model on realistic video footage. Therefore, we implemented the model in PyTorch and tested the model on two example videos.

Fashion Landmark Estimation with HRNet

Do Neural Networks for Segmentation Understand Insideness?

Supporting code for the Neograd algorithm

Python framework for Stochastic Differential Equations modeling

Self-Guided Contrastive Learning for BERT Sentence Representations

Implementation for "Domain-Specific Bias Filtering for Single Labeled Domain Generalization"

Migration of Edge-based Distributed Federated Learning