Testing and summary video generation

nchucvml / STVT

Video Summarization With Spatiotemporal Vision Transformer

Apache License 2.0

17 stars 7 forks source link

Closed Vaibhav05kansal closed 2 months ago

Vaibhav05kansal commented 2 months ago

We are able to train the model and .pth file is getting generated. How can we use this model on a given h5 file and generate the output video??

nchucvml commented 2 months ago

Please modify to val(model, val_loader, epoch, args) in train.py to get the test results.