cppe5_use_data_finetuning
yoon0915
Detección de objetos
Este modelo es una versión ajustada de facebook/detr-resnet-50 en el conjunto de datos cppe-5.
Como usar
Los siguientes hiperparámetros se usaron 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: 100
Las versiones de los marcos usados son:
Transformers 4.34.1
Pytorch 2.1.0+cu118
Datasets 2.14.6
Tokenizers 0.14.1
Funcionalidades
- Detección de objetos
- Transformers
- PyTorch
Casos de uso
- Este modelo puede ser utilizado para detectar objetos en imágenes y videos usando el conjunto de datos cppe-5. Puede ser aplicado en la automatización de tareas de detección de objetos en diversas industrias.