detr-r50-cd45rb-2ah-6l-corrected

polejowska
Detección de objetos

Este modelo es una versión ajustada de facebook/detr-resnet-50 en el conjunto de datos cd45rb_nan_xywh. Alcanza los siguientes resultados en el conjunto de evaluación: Loss: 1.7676.

Como usar

El modelo ha sido ajustado utilizando los siguientes hiperparámetros durante el 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: lineal
num_epochs: 25
mixed_precision_training: AMP Nativo

Resultados del entrenamiento (Pérdida de validación):

2.4426 (época 1)
2.1671 (época 2)
2.0925 (época 3)
2.0215 (época 4)
1.9689 (época 5)
1.9528 (época 6)
1.9058 (época 7)
1.8889 (época 8)
1.8751 (época 9)
1.8623 (época 10)
1.8636 (época 11)
1.8500 (época 12)
1.8354 (época 13)
1.8350 (época 14)
1.8243 (época 15)
1.8304 (época 16)
1.8243 (época 17)
1.8161 (época 18)
1.8106 (época 19)
1.7949 (época 20)
1.7747 (época 21)
1.7796 (época 22)
1.7757 (época 23)
1.7746 (época 24)
1.7676 (época 25)

Funcionalidades

Detección de objetos
Uso de Transformers
Implementado en PyTorch
Generado a partir de Trainer
Soporte de puntos de inferencia

Casos de uso

Detección de objetos en imágenes
Aplicaciones de vigilancia y seguridad
Análisis de imágenes para diagnóstico médico
Monitoreo automatizado en fábricas