https://github.com/abdulhadi806/gun-detection
Yolov8 model finetuned on hands gun dataset on L4 GPUs
https://github.com/abdulhadi806/gun-detection
ai computervision gun-detection opencv yolov8
Last synced: 5 months ago
JSON representation
Yolov8 model finetuned on hands gun dataset on L4 GPUs
- Host: GitHub
- URL: https://github.com/abdulhadi806/gun-detection
- Owner: AbdulHadi806
- License: mit
- Created: 2024-09-10T15:16:04.000Z (almost 2 years ago)
- Default Branch: main
- Last Pushed: 2024-10-09T19:41:37.000Z (over 1 year ago)
- Last Synced: 2025-02-12T02:09:32.503Z (over 1 year ago)
- Topics: ai, computervision, gun-detection, opencv, yolov8
- Language: Python
- Homepage:
- Size: 5.3 MB
- Stars: 0
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# Weapon Detection System
This project implements a real-time weapon detection system using the YOLOv8 model. The system is designed to detect and track weapons (specifically guns) within video streams or video files, providing a visual indication by highlighting detected weapons in each video frame.
## Model and Dataset
- **Model Used:** YOLOv8, a state-of-the-art object detection model known for its speed and accuracy.
- **Dataset:** The model is trained on a dataset containing 6,000 annotated images focused on hand-held weapons.
- **Dataset Link:** [Hand Weapon Dataset on Hugging Face](https://huggingface.co/datasets/AbdulHadi806/hand-weapone-dataset)
- **Dataset Creation Tool:** The dataset was curated and annotated using [Roboflow](https://roboflow.com/), a popular tool for generating high-quality datasets.
## Technology Stack
- **OpenCV:** Utilized for video processing, frame handling, and drawing bounding boxes around detected objects.
- **YOLOv8:** Employed for its efficient object detection capabilities, enabling the system to identify weapons with high accuracy.
## Installation
To get started, clone the repository and install the required dependencies:
```bash
git clone https://github.com/your-username/weapon-detection-system.git
cd weapon-detection-system
python main_area.py
```
