videomae-base-finetuned-ucf101_sport-subset
IsraelSonseca
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: 0.9609, Precisión: 0.7692.
Como usar
El modelo se puede utilizar implementando este código de ejemplo:
# Cargar el modelo
from transformers import VideoMAEForVideoClassification, VideoMAEFeatureExtractor
model = VideoMAEForVideoClassification.from_pretrained('IsraelSonseca/videomae-base-finetuned-ucf101_sport-subset')
feature_extractor = VideoMAEFeatureExtractor.from_pretrained('IsraelSonseca/videomae-base-finetuned-ucf101_sport-subset')
# Preprocesamiento del video
inputs = feature_extractor(videos, return_tensors='pt')
# Clasificación
outputs = model(**inputs)
logits = outputs.logits
Funcionalidades
- Clasificación de videos
- Transformadores
- PyTorch
- TensorBoard
Casos de uso
- Clasificación de videos deportivos