detr-r50-cd45rb-8ah-6l-gelu-corrected
polejowska
Detección de objetos
Este modelo es una versión afinada de facebook/detr-resnet-50 en el conjunto de datos cd45rb. Este modelo de detección de objetos utilizando Transformers en PyTorch ha sido generado a partir del Entrenador. Está bajo la licencia apache-2.0 y se ha logrado los siguientes resultados en el conjunto de evaluación: Pérdida: 1.5861.
Como usar
Para utilizar este modelo, puede usar los siguientes hiperparámetros de entrenamiento:
learning_rate: 1e-05
train_batch_size: 4
eval_batch_size: 8
seed: 42
optimizer: Adam (con betas=(0.9, 0.999) y epsilon=1e-08)
lr_scheduler_type: linear
num_epochs: 25
mixed_precision_training: Native AMP
A continuación se muestran los resultados del entrenamiento en diferentes épocas:
Epoch 1: Training Loss = 3.0438, Validation Loss = 1.9413
Epoch 2: Training Loss = 2.3933, Validation Loss = 1.8238
...
Epoch 25: Training Loss = 1.9975, Validation Loss = 1.5861
Funcionalidades
- Detección de objetos
- Transformers
- PyTorch
- Entrenamiento con precisiones mixtas
- Scheduler tipo lineal
Casos de uso
- Detección y reconocimiento de objetos en imágenes
- Aplicaciones de vigilancia y seguridad
- Sistemas de asistencia en la conducción
- Análisis de contenido visual para marketing