detr-resnet-50_finetuned_cppe5

hayatu
Detección de objetos

Este modelo es una versión afinada de facebook/detr-resnet-50 en el conjunto de datos cppe-5.

Como usar

La siguiente configuración de hiperparámetros fue utilizada 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

Versiones de Frameworks

  • Transformers 4.30.2
  • Pytorch 2.0.1+cu118
  • Datasets 2.13.0
  • Tokenizers 0.13.3

Funcionalidades

Detección de objetos
Transformadores
PyTorch
TensorBoard

Casos de uso

Detección de objetos en imágenes