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

Transfer Reinforcement Learning for Differing Action Spaces via Q-Network Representations

ALBERT-pytorch-implementation - ALBERT pytorch implementation

The lightweight PyTorch wrapper for high-performance AI research. Scale your models, not the boilerplate.

LAMDA: Label Matching Deep Domain Adaptation

A set of Deep Reinforcement Learning Agents implemented in Tensorflow.

A Comprehensive Analysis of Weakly-Supervised Semantic Segmentation in Different Image Domains (IJCV submission)

Bayesian optimisation library developped by Huawei Noah's Ark Library

A Comparative Review of Recent Kinect-Based Action Recognition Algorithms (TIP2020, Matlab codes)

DAFNe: A One-Stage Anchor-Free Deep Model for Oriented Object Detection

An open source machine learning library for performing regression tasks using RVM technique.

Implementation of the 😇 Attention layer from the paper, Scaling Local Self-Attention For Parameter Efficient Visual Backbones

Official implementation of "Learning Proposals for Practical Energy-Based Regression", 2021.

ChainerRL is a deep reinforcement learning library built on top of Chainer.

Code for the paper "Balancing Training for Multilingual Neural Machine Translation, ACL 2020"

Official TensorFlow code for the forthcoming paper

(Python, R, C/C++) Isolation Forest and variations such as SCiForest and EIF, with some additions (outlier detection + similarity + NA imputation)

BankNote-Net: Open dataset and encoder model for assistive currency recognition

Volumetric parameterization of the placenta to a flattened template

Discriminative Condition-Aware PLDA

We present a regularized self-labeling approach to improve the generalization and robustness properties of fine-tuning.