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