detr-r50-cd45rb-8ah-4l-1cost

El siguiente hiperparámetros se usaron durante el entrenamiento: learning_rate: 1e-06 train_batch_size: 4 eval_batch_size: 8 seed: 42 optimizer: Adam con betas=(0.9,0.999) y epsilon=1e-08 lr_scheduler_type: linear num_epochs: 20 mixed_precision_training: Native AMP

import torch
from transformers import DetrForObjectDetection, DetrImageProcessor

# Cargar el modelo y procesador
model = DetrForObjectDetection.from_pretrained('polejowska/detr-r50-cd45rb-8ah-4l-1cost')
processor = DetrImageProcessor.from_pretrained('polejowska/detr-r50-cd45rb-8ah-4l-1cost')

# Preparar la entrada
inputs = processor(images=image, return_tensors='pt')

# Hacer predicciones
outputs = model(**inputs)