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

This is a collection of our NAS and Vision Transformer work.

DvD-TD3: Diversity via Determinants for TD3 version

Reinforcement Learning for the Blackjack

3D Avatar Lip Syncronization from speech (JALI based face-rigging)

Group Fisher Pruning for Practical Network Compression(ICML2021)

This repository contains code from the paper "TTS-GAN: A Transformer-based Time-Series Generative Adversarial Network"

StyleSwin: Transformer-based GAN for High-resolution Image Generation

Python based framework for Automatic AI for Regression and Classification over numerical data.

RLBot Python bindings for the Rust crate rl_ball_sym

A CV toolkit for my papers.

Lex Rosetta: Transfer of Predictive Models Across Languages, Jurisdictions, and Legal Domains

Google Landmark Recogntion and Retrieval 2021 Solutions

COVINS -- A Framework for Collaborative Visual-Inertial SLAM and Multi-Agent 3D Mapping

[ICCV 2021] Our work presents a novel neural rendering approach that can efficiently reconstruct geometric and neural radiance fields for view synthesis.

A Collection of LiDAR-Camera-Calibration Papers, Toolboxes and Notes

Official implementation of Rich Semantics Improve Few-Shot Learning (BMVC, 2021)

Unified file system operation experience for different backend

CVPR 2022 "Online Convolutional Re-parameterization"

Gesture-controlled Video Game. Just swing your finger and play the game without touching your PC

AdaMML: Adaptive Multi-Modal Learning for Efficient Video Recognition