Skip to content

Latest commit

 

History

History
11 lines (9 loc) · 318 Bytes

README.md

File metadata and controls

11 lines (9 loc) · 318 Bytes

vidlab

Video frames classification using transformer model Put all videos in all_data folder

from features import get_features
from model import run_experiment

train_x, train_y, test_x, test_y, label_pr = get_features('all_data')
model = run_experiment(train_x, train_y, test_x, test_y, label_pr, 6)