urbanManul/furniture_use_data_finetuning_depre

urbanManul
Detección de objetos

Este modelo es una versión afinada de facebook/detr-resnet-50 en el conjunto de datos cppe-5.

Como usar

# Uso del modelo para inferencia:

from transformers import DetrFeatureExtractor, DetrForObjectDetection
import torch

# Cargar el modelo y el extractor de características
feature_extractor = DetrFeatureExtractor.from_pretrained('urbanManul/furniture_use_data_finetuning_depre')
model = DetrForObjectDetection.from_pretrained('urbanManul/furniture_use_data_finetuning_depre')

# Obtener las imágenes
images = [...] # Lista de imágenes
inputs = feature_extractor(images=images, return_tensors='pt')

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

# Proceso de salida
logits = outputs.logits
boxes = outputs.pred_boxes

Funcionalidades

Detección de objetos
Transformers
PyTorch

Casos de uso

Detección de uso de muebles en imágenes
Optimización en sistemas de seguridad
Identificación y clasificación de objetos en el entorno