keremberke/hard-hat-detection
Viewer • Updated • 19.8k • 557 • 11
How to use lanseria/yolov8n-hard-hat-detection_web_model with ultralytics:
from ultralytics import YOLOvv8
model = YOLOvv8.from_pretrained("lanseria/yolov8n-hard-hat-detection_web_model")
source = 'http://images.cocodataset.org/val2017/000000039769.jpg'
model.predict(source=source, save=True)This model is built using tfjs and is based on the YOLOv8n architecture. It is capable of detecting two classes of objects: people wearing safety helmets and those who are not.
该模型使用tfjs构建,基于YOLOv8n架构,可以检测两类物体:戴安全帽的人和未戴安全帽的人。
This model is converted from https://huggingface.co/keremberke/yolov8n-hard-hat-detection
该模型转换自 https://huggingface.co/keremberke/yolov8n-hard-hat-detection
["Hardhat", "NO-Hardhat"]