modelo_entrenado
LiviaQi
Detección de objetos
Este modelo es una versión ajustada de facebook/detr-resnet-50 en el conjunto de datos imagefolder.
Como usar
from transformers import DetrFeatureExtractor, DetrForObjectDetection
from PIL import Image
import requests
url = 'URL_DE_LA_IMAGEN'
image = Image.open(requests.get(url, stream=True).raw)
feature_extractor = DetrFeatureExtractor.from_pretrained('facebook/detr-resnet-50')
model = DetrForObjectDetection.from_pretrained('LiviaQi/trained_model')
inputs = feature_extractor(images=image, return_tensors='pt')
outputs = model(**inputs)
# Procesar las salidas
Funcionalidades
- Detección de objetos
- Transformers
- PyTorch
- TensorBoard
- imagefolder
- detr
Casos de uso
- Detección de objetos en imágenes
- Análisis de imágenes
- Automatización de etiquetado de imágenes