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