detr-resnet-50_finetuned_cppe5
Shebrain
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
El modelo fue entrenado con los siguientes hiperparámetros:
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
Versiones de framework utilizadas:
- Transformers 4.25.1
- Pytorch 1.13.1+cu116
- Datasets 2.8.0
- Tokenizers 0.13.2
Funcionalidades
- Detección de objetos
- Transformadores
- PyTorch
Casos de uso
- El modelo puede ser usado para tareas de detección de objetos en imágenes.
- Puede ser integrado en aplicaciones que requieren identificar múltiples objetos dentro de una imagen.