detr-r50-cd45rb-2ah-6l

polejowska

Detección de objetos

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

Como usar

Los siguientes hiperparámetros fueron utilizados 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: 5
mixed_precision_training: Native AMP

Funcionalidades

Detección de objetos
Transformers
Entrenado con PyTorch

Casos de uso

Detección de objetos en imágenes