videomae-base-finetuned-ucf101-subset-SBDtoy
HaileyJu
Clasificación de video
Este modelo es una versión afinada de MCG-NJU/videomae-base en un conjunto de datos desconocido. Logra los siguientes resultados en el conjunto de evaluación: Pérdida: 1.8346, Exactitud: 0.6275.
Como usar
Para usar el modelo, puedes cargarlo y ejecutar la inferencia utilizando las librerías de Transformers y PyTorch. Aquí hay un ejemplo de cómo hacerlo:
from transformers import VideoMAEModel, VideoMAEFeatureExtractor
import torch
# Cargar el modelo y el extractor de características
model = VideoMAEModel.from_pretrained('HaileyJu/videomae-base-finetuned-ucf101-subset-SBDtoy')
feature_extractor = VideoMAEFeatureExtractor.from_pretrained('HaileyJu/videomae-base-finetuned-ucf101-subset-SBDtoy')
# Preparar un video de entrada
video = ... # cargar tu video aquí
inputs = feature_extractor(video, return_tensors='pt')
# Realizar la inferencia
with torch.no_grad():
outputs = model(**inputs)
logits = outputs.logits
# Procesar los resultados
... # tu código de post-procesamiento aquí
Funcionalidades
- Clasificación de video
- Transformers
- PyTorch
- Safetensors
- Generado a partir de Trainer
Casos de uso
- Clasificación de videos educativos
- Análisis de seguridad en videos de vigilancia
- Etiquetado automático de videos