Skip to content

PPWangyc/rvt

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Recurrent Video Transformer(RVT)

Environment

pytorch(1.12.1), sklearn, numpy, python 3.8+

Training

  1. Save the pre-trained model under the checkpoint folder, Download here
  2. Place the Dataset and adjust the script path variable.
  3. Go to the scripts folder
source train_rvt.sh

The script will automatically implement the Leave One Out Cross-Validation(LOOCV). The running log file train.log will be output in log folder. The experiment results are output in output_rvt.json in output folder

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors