detr-resnet-50_finetuned_cppe5

import torch
from transformers import DetrForObjectDetection, DetrImageProcessor
from PIL import Image
import requests

# Cargar el modelo y el procesador de imágenes
model = DetrForObjectDetection.from_pretrained('Mustafa21/detr-resnet-50_finetuned_cppe5')
processor = DetrImageProcessor.from_pretrained('Mustafa21/detr-resnet-50_finetuned_cppe5')

# Obtener una imagen de ejemplo
url = 'http://imagen_url_ejemplo.com'
image = Image.open(requests.get(url, stream=True).raw)

# Procesar la imagen
inputs = processor(images=image, return_tensors='pt')

# Realizar la inferencia
outputs = model(**inputs)

# Procesar los resultados
boxes, scores, labels = outputs.get('boxes'), outputs.get('scores'), outputs.get('labels')