Model Details
YoloV8n 3M parameters model for Guns Detection, trained on 16k images
Model Description
- Shared by [optional]: Karun Sharma
- License: MIT
Model Sources [optional]
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.
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