How to train the model?

rpm1412 / LUS_Video_Summarization

An implementation of the summarization algorithm in the paper titled - Unsupervised multi-latent space reinforcement learning framework for video summarization in ultrasound imaging.

MIT License

7 stars 3 forks source link

How to train the model? #1

Closed zhw2020913 closed 2 years ago

zhw2020913 commented 2 years ago

Can you please tell me how to train the model onmy own dataset ?

rpm1412 commented 2 years ago

Hello, Please find the references to open source repositories for deep reinforcement learning for unsupervised video summarization cited in the manuscript : 1) https://github.com/KaiyangZhou/pytorch-vsumm-reinforce 2) https://github.com/Lorna-Liu/ultrasound_vsumm_RL

Thanks !

zhw2020913 commented 2 years ago

ok，thanks.