search

li-plus / DSNet

DSNet: A Flexible Detect-to-Summarize Network for Video Summarization
https://ieeexplore.ieee.org/document/9275314
MIT License
209 stars 50 forks source link

pre-trained model #15

Closed Bensong0506 closed 3 years ago

Bensong0506 commented 3 years ago

Hi,

May I ask why there are five .pt file for each datasets in the pre-trained model. I have tried the model on the default customer video and all five file provides quite different results. Which one should I choose? Thanks in advance.

li-plus commented 3 years ago

Because we train our model 5 times on each dataset config, and each time the dataset is randomly split into training & validation sets. The dataset is too small to train a stable model. Maybe you could train it on a larger dataset.