Find Objects and Focus on Highlights: Mining Object Semantics for Video Highlight Detection via Graph Neural Networks
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
Fill dataset, feat_dir, gpu in the config_server.py
python train.py
python eval.py