detr-resnet-50_finetuned_cppe5

# Ejemplo de uso del modelo:
from transformers import DetrForObjectDetection, DetrFeatureExtractor
import torch
from PIL import Image
import requests

# Cargar el modelo y el feature extractor
model = DetrForObjectDetection.from_pretrained('liuliu96/detr-resnet-50_finetuned_cppe5')
feature_extractor = DetrFeatureExtractor.from_pretrained('liuliu96/detr-resnet-50_finetuned_cppe5')

# Cargar una imagen de ejemplo
url = 'https://example.com/your-image.jpg'
image = Image.open(requests.get(url, stream=True).raw)

# Preprocesar la imagen
inputs = feature_extractor(images=image, return_tensors='pt')

# Realizar la detección de objetos
outputs = model(**inputs)

# Acceder a las predicciones
logits = outputs.logits
boxes = outputs.pred_boxes