detr-finetuned-boat-detection
nielsr
Detección de objetos
Un modelo de detección de objetos utilizando transformadores específicamente ajustado para la detección de barcos. Este modelo emplea la arquitectura DETR (Detractor) y está afinado para identificar barcos en imágenes. Usa PyTorch y Safetensors para realizar inferencias.
Como usar
Para usar este modelo, puede implementarlo en sus propios sistemas de detección de objetos con PyTorch. A continuación se muestra cómo puedes cargar y utilizar el modelo en un script de Python:
import torch
from transformers import DetrForObjectDetection, DetrFeatureExtractor
model_name = 'nielsr/detr-finetuned-boat-detection'
model = DetrForObjectDetection.from_pretrained(model_name)
feature_extractor = DetrFeatureExtractor.from_pretrained(model_name)
# Cargar una imagen
image = ... # tu imagen aquí
# Preprocesar la imagen
inputs = feature_extractor(images=image, return_tensors='pt')
# Hacer predicción
outputs = model(**inputs)
# Procesar la salida
logits = outputs.logits
boxes = outputs.pred_boxes
Funcionalidades
- Detección de objetos
- Arquitectura de transformadores DETR
- Optimizado para la detección de barcos
- Compatible con PyTorch
- Utiliza Safetensors
Casos de uso
- Detección de barcos en imágenes satelitales
- Monitoreo de actividades marítimas
- Reconocimiento de embarcaciones en tiempo real
- Aplicaciones en seguridad marítima