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