detr-r50-cd45rb-8ah-4l-corrected
polejowska
Detección de objetos
Este modelo es una versión ajustada de facebook/detr-resnet-50 en el conjunto de datos cd45rb. Alcanzó los siguientes resultados en el conjunto de evaluación: Pérdida: 1.8325.
Como usar
El modelo fue entrenado usando los siguientes hiperparámetros:
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: 10
mixed_precision_training: AMP Nativo
Resultados del entrenamiento:
Pérdida de Entrenamiento
Época
Paso
Pérdida de Validación
3.0619
1.0
4606
2.1934
2.6954
2.0
9212
2.0490
2.5695
3.0
13818
1.9843
2.518
4.0
18424
1.9648
2.4652
5.0
23030
1.9354
2.4235
6.0
27636
1.9127
2.3947
7.0
32242
1.8715
2.369
8.0
36848
1.8564
2.3542
9.0
41454
1.8511
2.3407
10.0
46060
1.8325
Versiones del framework:
Transformers 4.28.0
Pytorch 2.0.1
Datasets 2.12.0
Tokenizers 0.13.3
Funcionalidades
- Detección de objetos
- Transformers
- PyTorch
Casos de uso
- Detección automática de objetos en imágenes
- Aplicaciones que requieren reconocimiento de objetos
- Integración en sistemas de visión por computadora