To be uploaded soon
Just run the scripts in the bin
folder.
train_dahlia_rgb.sh
: Train I3D on Dahlia RGB
train_dahlia_skeletons.sh
: Train LSTM on Dahlia Skeletons
train_pku_rgb.sh
: Train I3D on PKU-MMD RGB
train_pku_skeletons
: Train LSTM on PKU-MMD Skeletons
dataset_name
:pku-mmd
ordahlia
model
:i3d
orlstm
num_epochs
: Number of epochsbatch_size
: Batch size for trainingeval_batch_size
: Batch size for evaluationexperiment_name
: Name of the experimentlog_dir
: Directory to keep all the result innum_gpus
: Number of GPUs
dahlia_train_path
: Path to the training filedahlia_validation_path
: Path to the validation filedahlia_test_path
: Path to the testing filedahlia_skeletons_dir
: Directory to get the skeletos fromdahlia_modality
:rgb
orskeletons
pku_mmd_split
:cross_view
orcross_subject
pku_mmd_rgb_dir
: Directory to get the RGB frames frompku_mmd_skeletons_dir
: Directory to get the skeletons frompku_mmd_labels_dir
: Directory to get the label files frompku_mmd_splits_dir'
: Directory to get the split file frompku_mmd_modality
:rgb
orskeletons
dropout
: Dropouttime_steps
: Number of time stepsnum_neurons
: Number of neuronsdata_size
: Size of the input feature vector