detr-resnet-50_finetuned_cppe5
miriaiml
Detección de objetos
Este modelo es una versión afinada de facebook/detr-resnet-50 en el dataset cppe-5.
Como usar
Los siguientes hiperparámetros fueron usados 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
Las versiones del framework usadas son las siguientes:
- Transformers 4.34.1
- Pytorch 2.1.0+cu118
- Datasets 2.14.5
- Tokenizers 0.14.1
Este modelo no tiene suficiente actividad para ser desplegado en la Inference API (sin servidor) aún. Incrementa su visibilidad social y revisa más tarde, o despliega en los Endpoints de Inferencia (dedicados) en su lugar.
Funcionalidades
- Detección de objetos
- Transformadores
- PyTorch
Casos de uso
- Detección de objetos en imágenes usando el dataset cppe-5