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

polejowska

Detección de objetos

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

Como usar

Los siguientes hiperparámetros se usaron 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: linear
num_epochs: 10
mixed_precision_training: Native AMP

Funcionalidades

Modelo de detección de objetos
Basado en Transformers
Utiliza PyTorch
Optimizador: Adam con betas=(0.9, 0.999) y epsilon=1e-08
Tipo de programador de tasa de aprendizaje: lineal
Entrenamiento de precisión mixta: AMP nativo

Casos de uso

Detección de objetos en imágenes
Aplicaciones de visión por computadora