videomae-base-finetuned-ucf101-subset
X-X-512
Clasificación de video
Este modelo es una versión ajustada de MCG-NJU/videomae-base en un conjunto de datos desconocido. Alcanza los siguientes resultados en el conjunto de evaluación: Pérdida: 0.4636, Precisión: 0.8581
Como usar
from transformers import VideoMAEForVideoClassification
# Cargar el modelo fine-tuned
model = VideoMAEForVideoClassification.from_pretrained('X-X-512/videomae-base-finetuned-ucf101-subset')
# Seleccionar el optimizador
optimizer = torch.optim.Adam(
model.parameters(), lr=5e-05, betas=(0.9, 0.999), eps=1e-08
)
# Configurar el planificador de tasa de aprendizaje
scheduler = torch.optim.lr_scheduler.LinearLR(
optimizer, warmup_ratio=0.1
)
Funcionalidades
- Clasificación de Video
- Transformers
- PyTorch
- TensorBoard
Casos de uso
- Clasificación de actividades en videos
- Reconocimiento de acciones en videos deportivos
- Aplicaciones en videovigilancia y seguridad