yolos-tiny-Detección-Sombrero-Seguridad

DunnBC22
Detección de objetos

Este modelo es una versión afinada de hustvl/yolos-tiny en el conjunto de datos de detección de casco de seguridad. Está diseñado para demostrar mi capacidad para resolver un problema complejo utilizando tecnología. Para más información sobre cómo fue creado, consulte el siguiente enlace: [Proyecto de Detección de Casco](https://github.com/DunnBC22/Vision_Audio_and_Multimodal_Projects/blob/main/Computer%20Vision/Object%20Detection/Hard%20Hat%20Detection/Hard_Hat_Object_Detection_YOLOS.ipynb)

Como usar

Procedimiento de entrenamiento

Los siguientes hiperparámetros se utilizaron durante el entrenamiento:

  • learning_rate: 5e-05
  • train_batch_size: 8
  • eval_batch_size: 8
  • seed: 42
  • optimizer: Adam con betas=(0.9, 0.999) y epsilon=1e-08
  • lr_scheduler_type: linear
  • num_epochs: 8

Resultados del entrenamiento

  • Precisión Media (AP) IoU=0.50:0.95 todos maxDets=100 0.346
  • Precisión Media (AP) IoU=0.50 todos maxDets=100 0.747
  • Precisión Media (AP) IoU=0.75 todos maxDets=100 0.275
  • Precisión Media (AP) IoU=0.50:0.95 pequeño maxDets=100 0.128
  • Precisión Media (AP) IoU=0.50:0.95 mediano maxDets=100 0.343
  • Precisión Media (AP) IoU=0.50:0.95 grande maxDets=100 0.521
  • Recuerdo Medio (AR) IoU=0.50:0.95 todos maxDets=1 0.188
  • Recuerdo Medio (AR) IoU=0.50:0.95 todos maxDets=10 0.484
  • Recuerdo Medio (AR) IoU=0.50:0.95 todos maxDets=100 0.558
  • Recuerdo Medio (AR) IoU=0.50:0.95 pequeño maxDets=100 0.320
  • Recuerdo Medio (AR) IoU=0.50:0.95 mediano maxDets=100 0.538
  • Recuerdo Medio (AR) IoU=0.50:0.95 grande maxDets=100 0.743

Versiones del marco de trabajo

  • Transformers 4.31.0
  • Pytorch 2.0.1+cu118
  • Datasets 2.14.3
  • Tokenizers 0.13.3

Funcionalidades

Detección de objetos
Transformers
Basado en PyTorch
Compatible con TensorBoard
Generado con Trainer

Casos de uso

Demostración de capacidad para resolver problemas complejos
Aumento de la seguridad en el lugar de trabajo detectando cascos de seguridad
Implementación en sistemas para vigilancia y monitoreo