--- 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 - **Shared by [optional]:** Karun Sharma - **License:** MIT ### Model Sources [optional] - **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 ```