koya1/videomae-base-finetuned-ucf101-subset

Este modelo está listo para ser utilizado para tareas de clasificación de videos. Aquí tienes un ejemplo del uso del modelo:

import torch
from transformers import VideoMAEForVideoClassification, VideoMAEFeatureExtractor

model = VideoMAEForVideoClassification.from_pretrained('koya1/videomae-base-finetuned-ucf101-subset')
feature_extractor = VideoMAEFeatureExtractor.from_pretrained('koya1/videomae-base-finetuned-ucf101-subset')

# Cargando un video
video = ...  # Asegúrate de que es un video tensor adecuado

# Extrayendo características
tokenized_video = feature_extractor(video, return_tensors='pt')

# Realizando una clasificación
outputs = model(**tokenized_video)

# Resultados
logits = outputs.logits
predicciones = torch.argmax(logits, dim=-1)
print('Predicciones:', predicciones)