object_detection_test_1

whyoke

Detección de objetos

Este modelo es una versión ajustada de facebook/detr-resnet-50 en el conjunto de datos cppe-5. Se utiliza para la detección de objetos y se entrena con Transformers, PyTorch y TensorBoard.

Como usar

Para utilizar este modelo, necesita los siguientes hiperparámetros durante el entrenamiento:
learning_rate: 1e-05
train_batch_size: 2
eval_batch_size: 8
seed: 42
optimizer: Adam con betas=(0.9,0.999) y epsilon=1e-08
lr_scheduler_type: linear
num_epochs: 5
mixed_precision_training: Native AMP

Utiliza las siguientes versiones de frameworks:
Transformers 4.26.1
Pytorch 1.13.0
Datasets 2.10.1
Tokenizers 0.13.2

Funcionalidades

Detección de objetos
Transformers
PyTorch
TensorBoard

Casos de uso

Detección de objetos en imágenes