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

ICLR2021 (Under Review)

Based on Stockfish neural network(similar to LcZero)

This is an open-source toolkit for Heterogeneous Graph Neural Network(OpenHGNN) based on DGL [Deep Graph Library] and PyTorch.

The PyTorch improved version of TPAMI 2017 paper: Face Alignment in Full Pose Range: A 3D Total Solution.

Code for Pose-Controllable Talking Face Generation by Implicitly Modularized Audio-Visual Representation (CVPR 2021)

A pytorch-based real-time segmentation model for autonomous driving

Symmetry and Uncertainty-Aware Object SLAM for 6DoF Object Pose Estimation

Source code of the paper PatchGraph: In-hand tactile tracking with learned surface normals.

Disease Informed Neural Networks (DINNs) — neural networks capable of learning how diseases spread, forecasting their progression, and finding their unique parameters (e.g. death rate).

Pytorch Implementation of LNSNet for Superpixel Segmentation

[ICRA 2022] CaTGrasp: Learning Category-Level Task-Relevant Grasping in Clutter from Simulation

Driller: augmenting AFL with symbolic execution!

particle tracking model, works with the ROMS output file(qck.nc, his.nc)

Implementations for the ICLR-2021 paper: SEED: Self-supervised Distillation For Visual Representation.

Can we learn gradients by Hamiltonian Neural Networks?

Source code for our paper "Molecular Mechanics-Driven Graph Neural Network with Multiplex Graph for Molecular Structures"

HODEmu, is both an executable and a python library that is based on Ragagnin 2021 in prep.

[ICLR 2021, Spotlight] Large Scale Image Completion via Co-Modulated Generative Adversarial Networks

This repo tries to recognize faces in the dataset you created

A `Neural = Symbolic` framework for sound and complete weighted real-value logic