detr-resnet-50_finetuned_cppe5
emre
Detección de objetos
Este modelo es una versión ajustada de facebook/detr-resnet-50 en el conjunto de datos cppe-5.
Como usar
Los siguientes hiperparámetros se utilizaron durante el entrenamiento:
learning_rate: 1e-05
train_batch_size: 8
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
Resultado del entrenamiento:
Step Training Loss
300 2.162200
600 2.011000
1200 1.779500
Funcionalidades
- Detección de objetos
- Transformers
- PyTorch
- TensorBoard
Casos de uso
- Detectar objetos en imágenes usando una red neuronal profunda.
- Implementar modelos de detección de objetos en aplicaciones basadas en inteligencia artificial.