fb-detr-table_detection_v1.0
paturi1710
Detección de objetos
Este modelo es una versión ajustada de facebook/detr-resnet-50 en el conjunto de datos imagefolder. Logra los siguientes resultados en el conjunto de evaluación: Pérdida: 0.2380.
Como usar
El siguiente modelo se puede utilizar para la detección de tablas en imágenes.
from transformers import DetrForObjectDetection, DetrImageProcessor
import torch
from PIL import Image
import requests
# Cargar el modelo
model = DetrForObjectDetection.from_pretrained('paturi1710/fb-detr-table_detection_v1.0')
processor = DetrImageProcessor.from_pretrained('paturi1710/fb-detr-table_detection_v1.0')
# Procesar la imagen
url = 'URL_DE_LA_IMAGEN'
image = Image.open(requests.get(url, stream=True).raw)
inputs = processor(images=image, return_tensors='pt')
# Hacer la inferencia
with torch.no_grad():
outputs = model(**inputs)
# Obtener las predicciones
print(outputs)
Se puede ajustar aún más el modelo o usarlo tal cual para la detección de objetos en nuevas imágenes.
Funcionalidades
- Detección de objetos
- Transformers
- PyTorch
- TensorBoard
Casos de uso
- Detección de tablas en documentos escaneados
- Detección de objetos en imágenes para sistemas de gestión de contenido
- Automatización de la extracción de datos en imágenes para análisis