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

Emotion classification of online comments based on RNN

E-RAFT: Dense Optical Flow from Event Cameras

Flaxformer: transformer architectures in JAX/Flax

Deploy optimized transformer based models on Nvidia Triton server

StyleGAN2-ada for practice

Official PyTorch implementation of "VITON-HD: High-Resolution Virtual Try-On via Misalignment-Aware Normalization" (CVPR 2021)

Weakly-Supervised Semantic Segmentation Network with Deep Seeded Region Growing (CVPR 2018).

PASTRIE: A Corpus of Prepositions Annotated with Supersense Tags in Reddit International English

Breast Cancer Classification Model is applied on a different dataset

Weakly-supervised object detection.

A PyTorch library for Vision Transformers

Pytorch implementation of

This MVP data web app uses the Streamlit framework and Facebook's Prophet forecasting package to generate a dynamic forecast from your own data.

wgan, wgan2(improved, gp), infogan, and dcgan implementation in lasagne, keras, pytorch

On Evaluation Metrics for Graph Generative Models

Pytorch implementation of Deep Recursive Residual Network for Super Resolution (DRRN)

Code for 2021 NeurIPS --- Towards Multi-Grained Explainability for Graph Neural Networks

Migration of Edge-based Distributed Federated Learning

Pytorch Implementation for (STANet+ and STANet)

We evaluate our method on different datasets (including ShapeNet, CUB-200-2011, and Pascal3D+) and achieve state-of-the-art results, outperforming all the other supervised and unsupervised methods and 3D representations, all in terms of performance, accuracy, and training time.