detr-resnet-50_finetuned_cppe5
AladarMezga
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
# Cargar el modelo
from transformers import DetrForObjectDetection, DetrImageProcessor
# Especificar el modelo
model = DetrForObjectDetection.from_pretrained('AladarMezga/detr-resnet-50_finetuned_cppe5')
# Procesar las imágenes
processor = DetrImageProcessor.from_pretrained('AladarMezga/detr-resnet-50_finetuned_cppe5')
inputs = processor(images=image, return_tensors="pt")
# Realizar la detección
outputs = model(**inputs)
Funcionalidades
- Detección de objetos
- Transformadores
- PyTorch
- TensorBoard
Casos de uso
- Identificación de objetos en imágenes
- Aplicaciones de seguridad
- Análisis de imágenes médicas
- Automatización industrial