Object Detection
ultralytics
PyTorch
detr
ultralyticsplus
yolov8
yolo
vision
awesome-yolov8-models
Eval Results (legacy)
Instructions to use deyelive/detr-resnet-50-finetuned-construction-safety with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- ultralytics
How to use deyelive/detr-resnet-50-finetuned-construction-safety with ultralytics:
from ultralytics import YOLOvv8 model = YOLOvv8.from_pretrained("deyelive/detr-resnet-50-finetuned-construction-safety") source = 'http://images.cocodataset.org/val2017/000000039769.jpg' model.predict(source=source, save=True) - Notebooks
- Google Colab
- Kaggle
| tags: | |
| - ultralyticsplus | |
| - yolov8 | |
| - ultralytics | |
| - yolo | |
| - vision | |
| - object-detection | |
| - pytorch | |
| - awesome-yolov8-models | |
| library_name: ultralytics | |
| library_version: 8.0.23 | |
| inference: true | |
| datasets: | |
| - keremberke/protective-equipment-detection | |
| model-index: | |
| - name: deyelive/yolov8m-protective-equipment-detection | |
| results: | |
| - task: | |
| type: object-detection | |
| dataset: | |
| type: keremberke/protective-equipment-detection | |
| name: protective-equipment-detection | |
| split: validation | |
| metrics: | |
| - type: precision | |
| value: 0.27342 | |
| name: mAP@0.5(box) | |
| pipeline_tag: object-detection | |
| <div align="center"> | |
| <img width="640" alt="keremberke/yolov8m-protective-equipment-detection" src="https://huggingface.co/keremberke/yolov8m-protective-equipment-detection/resolve/main/thumbnail.jpg"> | |
| </div> | |
| ### Supported Labels | |
| ``` | |
| ['glove', 'goggles', 'helmet', 'mask', 'no_glove', 'no_goggles', 'no_helmet', 'no_mask', 'no_shoes', 'shoes'] | |
| ``` | |
| ### How to use | |
| - Install [ultralyticsplus](https://github.com/fcakyon/ultralyticsplus): | |
| ```bash | |
| pip install ultralyticsplus==0.0.24 ultralytics==8.0.23 | |
| ``` | |
| - Load model and perform prediction: | |
| ```python | |
| from ultralyticsplus import YOLO, render_result | |
| # load model | |
| model = YOLO('keremberke/yolov8m-protective-equipment-detection') | |
| # set model parameters | |
| model.overrides['conf'] = 0.25 # NMS confidence threshold | |
| model.overrides['iou'] = 0.45 # NMS IoU threshold | |
| model.overrides['agnostic_nms'] = False # NMS class-agnostic | |
| model.overrides['max_det'] = 1000 # maximum number of detections per image | |
| # set image | |
| image = 'https://github.com/ultralytics/yolov5/raw/master/data/images/zidane.jpg' | |
| # perform inference | |
| results = model.predict(image) | |
| # observe results | |
| print(results[0].boxes) | |
| render = render_result(model=model, image=image, result=results[0]) | |
| render.show() | |
| ``` | |
| **More models available at: [awesome-yolov8-models](https://yolov8.xyz)** |