pytorch-vsumm-reinforce
pytorch-vsumm-reinforce copied to clipboard
Extracting image features for videos
Hey @Petersteve! Were you able to extract the features for generating a summary for a custom video?
@mayank26saxena Did you test the model on custom videos?
Anyone who did it?
you can see here, I make some introduction code
First make datastes
python video_forward2.py --makedatasets --dataset data_our/data_h5/data1.h5 --video-dir data_video/data1/ --frm-dir data_our/frames
Second make score and generate summary
python3 video_forward2.py --makescore --model log/summe-split0/model_epoch1000.pth.tar --gpu 0 --dataset data_our/data_h5/data2.h5 --save-dir logs/videolog/ \ --summary --frm-dir data_our/frames
I trained the model and tested it . You can refer my repo https://github.com/anaghazachariah/video_summary_generaton