Fantast/Yolos-pequeño-ajustado-para-focas

Fantast

Detección de objetos

Modelo YOLOS (de tamaño pequeño) ajustado para la tarea de detección de focas basado en hustvl/yolos-small. Fine-tuned en nuestro conjunto de datos de imágenes de focas.

Como usar

Aquí se muestra cómo usar este modelo:
from transformers import YolosFeatureExtractor, YolosForObjectDetection
from PIL import Image
import requests

image = Image.open('xxxxxxxxxxxxx')

feature_extractor = YolosFeatureExtractor.from_pretrained('fantast/yolos-small-finetuned-for-seal')
model = YolosForObjectDetection.from_pretrained('fantast/yolos-small-finetuned-for-seal')

inputs = feature_extractor(images=image, return_tensors='pt')
outputs = model(**inputs)

logits = outputs.logits
bboxes = outputs.pred_boxes

Actualmente, tanto el extractor de características como el modelo soportan PyTorch.

Funcionalidades

Detección de objetos
Transformadores de visión
Soporte para PyTorch

Casos de uso

Detección automática de focas en imágenes