|
|
---
|
|
|
license: mit
|
|
|
language:
|
|
|
- en
|
|
|
metrics:
|
|
|
- precision
|
|
|
pipeline_tag: object-detection
|
|
|
tags:
|
|
|
- Classification
|
|
|
- YoloV8
|
|
|
- Guns Detection
|
|
|
---
|
|
|
|
|
|
## Model Details
|
|
|
YoloV8n 3M parameters model for Guns Detection, trained on 16k images
|
|
|
### Model Description
|
|
|
|
|
|
<!-- Provide a longer summary of what this model is. -->
|
|
|
|
|
|
|
|
|
|
|
|
- **Shared by [optional]:** Karun Sharma
|
|
|
- **License:** MIT
|
|
|
|
|
|
### Model Sources [optional]
|
|
|
|
|
|
<!-- Provide the basic links for the model. -->
|
|
|
|
|
|
- **Demo:** https://huggingface.co/spaces/Zcket/Gun_Detect
|
|
|
|
|
|
## Uses
|
|
|
Can be used to detect presenece of Guns in images for Moderation.
|
|
|
|
|
|
|
|
|
|
|
|
### Downstream Use [optional]
|
|
|
Classes of Arms/Guns(Pistols, Grenades)
|
|
|
|
|
|
|
|
|
## How to Get Started with the Model
|
|
|
|
|
|
Use the code below to get started with the model.
|
|
|
|
|
|
```py
|
|
|
from ultralytics import YOLO
|
|
|
|
|
|
model = YOLO('best.pt')
|
|
|
|
|
|
results = model(['im1.jpg', 'im2.jpg']) # return a list of Results objects
|
|
|
|
|
|
# Process results list
|
|
|
for result in results:
|
|
|
boxes = result.boxes # Boxes object for bounding box outputs
|
|
|
masks = result.masks # Masks object for segmentation masks outputs
|
|
|
keypoints = result.keypoints # Keypoints object for pose outputs
|
|
|
probs = result.probs # Probs object for classification outputs
|
|
|
result.show() # display to screen
|
|
|
result.save(filename='result.jpg') # save to disk
|
|
|
```
|
|
|
|
|
|
|
