-
Check
requirements.txtand install the necessary packages (there are very few!). -
Prepare the data (features and labels) in the
datadirectory. In this code, we will work with the THUMOS14 dataset. Features and labels are available here. Extract the downloaded file into thedatadirectory. After all, the file structure should be as follows:data |--tmp.txt |--thumos14 | |--thumos14_4state_label_1 | | |--video_test_0000004.npy | | |--... | | |--... | |--thumos14_features | | |--video_test_0000004.npy | | |--... | | |--... | |--thumos14_oracle_proposals.pkl | |--thumos14_v2.json -
Download the classifier here, and place it in the
thumos14_classifier_modeldirectory. -
[Optional] Download checkpoint.pt and place it in the
t_oad_modeldirectory. -
For training, run the following command:
bash train.shThis script will train the model, make proposals with the model, and evaluate the Hungarian F1 score of the model's predictions.
-
To test the given checkpoint, complete step 4 and run the following command:
bash test.shThis will result in a
53.2hungarian f1 score.
musicalOffering/ActionSwitch-release
Folders and files
| Name | Name | Last commit date | ||
|---|---|---|---|---|