detr-resnet-50_finetuned_cppe5
bunbohue
Detección de objetos
Este modelo es una versión ajustada de facebook/detr-resnet-50 en un conjunto de datos desconocido.
Como usar
El modelo se puede utilizar para la detección de objetos. A continuación se presentan los hiperparámetros utilizados 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
Las versiones de framework utilizadas fueron:
- Transformers 4.37.2
- Pytorch 2.1.0+cu121
- Datasets 2.17.1
- Tokenizers 0.15.2
Funcionalidades
- Detección de objetos
- Transformers
- TensorBoard
- Safetensors
Casos de uso
- Detección de objetos en imágenes