detr-r50-cd45rb-4ah-12l

polejowska

Detección de objetos

Este modelo es una versión afinada de facebook/detr-resnet-50 en el conjunto de datos cd45rb. Consigue los siguientes resultados en el conjunto de evaluación: Pérdida: 2.1298.

Como usar

El siguiente conjunto de hiperparámetros se utilizó durante el entrenamiento:
learning_rate: 1e-05
train_batch_size: 4
eval_batch_size: 8
seed: 42
optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
lr_scheduler_type: linear
num_epochs: 5
mixed_precision_training: Native AMP

Resultados del entrenamiento:
Pérdida de Entrenamiento
Época
Paso
Pérdida de Validación

3.6967
1.0
4606
2.7254

3.2687
2.0
9212
2.4867

3.0302
3.0
13818
2.3065

2.8744
4.0
18424
2.1857

2.7778
5.0
23030
2.1298

Funcionalidades

Detección de objetos
Transformers
Modelos generados a partir de Trainer
Puntos finales de inferencia

Casos de uso

Detección de objetos en imágenes