cdetr-cd45rb-s
polejowska
Detección de objetos
Este modelo es una versión afinada de microsoft/conditional-detr-resnet-50 en el dataset None. Logra los siguientes resultados en el conjunto de evaluación: Pérdida: 2.0136.
Como usar
Procedimiento de entrenamiento
Hiperparámetros de entrenamiento
Los siguientes hiperparámetros se utilizaron durante el entrenamiento:
- learning_rate: 0.0001
- train_batch_size: 4
- 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
Resultados del entrenamiento
Pérdida de entrenamiento
| Epoch | Step | Validation Loss |
|-------|------|----------------|
| 1.0 | 242 | 2.2146 |
| 2.0 | 484 | 2.1106 |
| 3.0 | 726 | 2.1456 |
| 4.0 | 968 | 2.0863 |
| 5.0 | 1210 | 2.0294 |
| 6.0 | 1452 | 2.0859 |
| 7.0 | 1694 | 2.0731 |
| 8.0 | 1936 | 2.1052 |
| 9.0 | 2178 | 2.0378 |
| 10.0 | 2420 | 2.0136 |
Versiones del framework
- Transformers 4.28.1
- Pytorch 2.0.0+cu118
- Datasets 2.12.0
- Tokenizers 0.13.3
Funcionalidades
- Detección de objetos
- Transformers
- PyTorch
- Generado a partir de Trainer
Casos de uso
- Implementación de detección de objetos en imágenes
- Aplicaciones de visión artificial que requieren detección precisa de objetos