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

Source code for paper "Deep Diffusion Models for Robust Channel Estimation", TBA.

2D&3D human pose estimation

TransGAN: Two Transformers Can Make One Strong GAN

All the code and files related to the MI-Lab of UE19CS305 course in sem 5

A paper using optimal transport to solve the graph matching problem.

Model-based reinforcement learning in TensorFlow

Linear algebra python - Number of operations and problems in Linear Algebra and Numerical Linear Algebra

Neuron Merging: Compensating for Pruned Neurons (NeurIPS 2020)

graph-theoretic framework for robust pairwise data association

Image Segmentation using U-Net, U-Net with skip connections and M-Net architectures

Contour-guided image completion with perceptual grouping (BMVC 2021 publication)

Self-Supervised Learning for Domain Adaptation on Point-Clouds

Code for: Gradient-based Hierarchical Clustering using Continuous Representations of Trees in Hyperbolic Space. Nicholas Monath, Manzil Zaheer, Daniel Silva, Andrew McCallum, Amr Ahmed. KDD 2019.

A Domain-Agnostic Benchmark for Self-Supervised Learning

Implementation of EMNLP 2017 Paper "Natural Language Does Not Emerge 'Naturally' in Multi-Agent Dialog" using PyTorch and ParlAI

Simple PyTorch implementations of Badnets on MNIST and CIFAR10.

Modular Gaussian Processes

CvT2DistilGPT2 is an encoder-to-decoder model that was developed for chest X-ray report generation.

This is a Tensorflow implementation of Learning to See in the Dark in CVPR 2018

Deep Learning & 3D Convolutional Neural Networks for Speaker Verification