detr-r50-cd45rb-8ah-6l-50q

polejowska
Detección de objetos

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

Como usar

Este modelo usa PyTorch y se ha entrenado utilizando AMP nativo. Los hiperparámetros usados durante el entrenamiento incluyen:

  • 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

Funcionalidades

Detección de objetos
Transformers
PyTorch

Casos de uso

Detección de objetos en imágenes