/GNN-VH

Primary LanguagePython

Find Objects and Focus on Highlights: Mining Object Semantics for Video Highlight Detection via Graph Neural Networks

Requirements

Python 3.6

torchvision==0.3.0

torch==1.1.0

scikit-image==0.15.0

opencv-python==4.1.0.25

imageio==2.5.0

dgl==0.3

Configuration

Fill dataset, feat_dir, gpu in the config_server.py

Training

python train.py

Testing

python eval.py