ppe_yolos_model

Eshwar14

Detección de objetos

Este modelo es una versión ajustada de facebook/detr-resnet-50 en el conjunto de datos cppe-5.

Como usar

# Parámetros de hiperentrenamiento:
learning_rate: 1e-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: 1
mixed_precision_training: Native AMP

# Versiones del marco:
Transformers 4.29.2
Pytorch 2.0.1+cu118
Datasets 2.12.0
Tokenizers 0.13.3

Funcionalidades

Detección de objetos
Transformadores
PyTorch
TensorBoard

Casos de uso

Detección de Equipos de Protección Personal (EPP)