gun_dtct

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.

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