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

Official PyTorch implementation of RobustNet (CVPR 2021 Oral)

Funnels: Exact maximum likelihood with dimensionality reduction.

[CVPR 2021] Unsupervised Degradation Representation Learning for Blind Super-Resolution

Detectron2 is FAIR's next-generation platform for object detection and segmentation.

[ACM MM 2021] Joint Implicit Image Function for Guided Depth Super-Resolution

mPose3D, a mmWave-based 3D human pose estimation model.

The implementation for paper Joint t-SNE for Comparable Projections of Multiple High-Dimensional Datasets.

Using VapourSynth with super resolution models and speeding them up with TensorRT.

RCD: Relation Map Driven Cognitive Diagnosis for Intelligent Education Systems

PyTorch-lightning implementation of the ESFW module proposed in our paper Edge-Selective Feature Weaving for Point Cloud Matching

Automated Attendance Project Using Face Recognition

JAXDL: JAX (Flax) Deep Learning Library

2021-MICCAI-Progressively Normalized Self-Attention Network for Video Polyp Segmentation

Pytorch0.4.1 codes for InsightFace

Self-Supervised Vision Transformers Learn Visual Concepts in Histopathology (LMRL Workshop, NeurIPS 2021)

Cross View SLAM

Collection of machine learning related notebooks to share.

TensorFlow ROCm port

Multi-Task Temporal Shift Attention Networks for On-Device Contactless Vitals Measurement (NeurIPS 2020)

Cervix ROI Segmentation Using U-NET