videomae-base-finetuned-ucf101-subset-finetuned
ali5341
Clasificación de video
Este modelo fue entrenado desde cero en un conjunto de datos desconocido. Alcanza los siguientes resultados en el conjunto de evaluación: Pérdida: 0.9147 Precisión: 0.7701 El modelo está diseñado para la clasificación de video utilizando transformadores y PyTorch.
Como usar
import torch
from transformers import VideoMAEFeatureExtractor, VideoMAEForVideoClassification
# Cargar el modelo y el extractor de características
extractor = VideoMAEFeatureExtractor.from_pretrained('ali5341/videomae-base-finetuned-ucf101-subset-finetuned')
model = VideoMAEForVideoClassification.from_pretrained('ali5341/videomae-base-finetuned-ucf101-subset-finetuned')
# Ejemplo de uso
video = load_your_video_function('path_to_your_video')
inputs = extractor(video, return_tensors='pt')
outputs = model(**inputs)
# Obtener las predicciones
predictions = torch.argmax(outputs.logits, dim=-1)
Funcionalidades
- Clasificación de video
- Transformadores
- PyTorch
- Generado a partir de Trainer
Casos de uso
- Clasificación automática de videos
- Análisis de contenido de video
- Aplicaciones de seguridad
- Etiquetado de videos