detr-resnet-50-finetuned-construction-safety
deyelive
Detección de objetos
Modelo de detección de objetos finetuneado para la seguridad en la construcción. Detecta equipos de protección como guantes, gafas, casco, mascarilla, entre otros.
Como usar
Instalación de ultralyticsplus:
pip install ultralyticsplus==0.0.24 ultralytics==8.0.23
Carga del modelo y realización de predicciones:
from ultralyticsplus import YOLO, render_result
# cargar modelo
model = YOLO('keremberke/yolov8m-protective-equipment-detection')
# configurar parámetros del modelo
model.overrides['conf'] = 0.25 # Umbral de confianza NMS
model.overrides['iou'] = 0.45 # Umbral IoU NMS
model.overrides['agnostic_nms'] = False # NMS sin clase
model.overrides['max_det'] = 1000 # número máximo de detecciones por imagen
# establecer imagen
image = 'https://github.com/ultralytics/yolov5/raw/master/data/images/zidane.jpg'
# realizar inferencia
results = model.predict(image)
# observar resultados
print(results[0].boxes)
render = render_result(model=model, image=image, result=results[0])
render.show()
Funcionalidades
- Detección de objetos
- Modelo finetuneado para la seguridad en la construcción
- Funciona con PyTorch
- Basado en ResNet-50
- Compatible con ultralytics
Casos de uso
- Detección de equipos de protección personal en sitios de construcción
- Garantizar el cumplimiento de las normas de seguridad
- Monitoreo automático de seguridad en el lugar de trabajo