obj_detection_5epochs

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

model_name = "wieheistdu/obj_detection_5epochs"
feature_extractor = DetrFeatureExtractor.from_pretrained(model_name)
model = DetrForObjectDetection.from_pretrained(model_name)

# Cargar una imagen de ejemplo
i = Image.open(requests.get('https://url/to/image', stream=True).raw)
inputs = feature_extractor(images=i, return_tensors="pt")
outputs = model(**inputs)
# Recuperar las predicciones
logits = outputs.logits
boxes = outputs.pred_boxes

print("Logits:", logits)
print("Boxes:", boxes)