detr-resnet-50_finetuned-room-objects

# Aplicación del modelo para la detección de objetos en imágenes.
# Ejemplo de código para cargar y usar el modelo.
from transformers import DetrForObjectDetection, DetrImageProcessor
import torch
from PIL import Image

# Cargar el modelo preentrenado y el procesador de imágenes
model = DetrForObjectDetection.from_pretrained('machinelearningzuu/detr-resnet-50_finetuned-room-objects')
processor = DetrImageProcessor.from_pretrained('machinelearningzuu/detr-resnet-50_finetuned-room-objects')

# Cargar una imagen
image = Image.open('path_to_image')
inputs = processor(images=image, return_tensors='pt')

# Realizar la detección de objetos
outputs = model(**inputs)
logits = outputs.logits
boxes = outputs.pred_boxes