-
Notifications
You must be signed in to change notification settings - Fork 4
/
Copy pathconfiguration.py
67 lines (58 loc) · 2.85 KB
/
configuration.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
# ----- NOT MODIFY THESE PATHS
input_folder = './Input'
C3D_path = './C3D_Features'
C3D_info_path = './C3D_Features_Info'
I3D_path = './I3D_Features'
trained_folder = './trained_models/train_'
score_path = './Scores'
all_ann_path = './training_splits/Sultani/test_annot310_sultani.txt'
filter_path = './Filters_FeatureMaps'
# ----- MODIFY THESE VARIABLES
use_lstm = False
use_i3d = False
## Weights of the pre-trained classifier (10 possibilities)
classifier_model_json = './trained_models/model.json'
#classifier_model_weigts = './trained_models/weights_L1L2.mat'
#classifier_model_weigts = './trained_models/train_exp1_C3D/weights_exp1_C3D.mat'
#classifier_model_weigts = './trained_models/train_exp2_C3D/weights_exp2_C3D.mat'
#classifier_model_weigts = './trained_models/train_exp1_128LSTM_C3D/weights_exp1_128LSTM_C3D.mat'
#classifier_model_weigts = './trained_models/train_exp2_128LSTM_C3D/weights_exp2_128LSTM_C3D.mat'
#classifier_model_weigts = './trained_models/train_exp1_256LSTM_C3D/weights_exp1_256LSTM_C3D.mat'
#classifier_model_weigts = './trained_models/train_exp1_I3D/weights_exp1_I3D.mat'
#classifier_model_weigts = './trained_models/train_exp2_I3D/weights_exp2_I3D.mat'
#classifier_model_weigts = './trained_models/train_sul_C3D/weights_sul_C3D.mat'
#classifier_model_weigts = './trained_models/train_sul_128LSTM_C3D/weights_sul_128LSTM_C3D.mat'
#classifier_model_weigts = './trained_models/train_sul_I3D/weights_sul_I3D.mat'
## Paths to use the entire UCF_Crimes dataset
path_all_videos = '.../UCF_Crimes/Videos/'
path_all_features = '.../UCF_Crimes/Videos/'
## Name any experiment you want to perform with train.py
train_exp_name = 'experiment_...'
## Choose which training-test set you prefer (3 possibilities)
sul = True
exp1 = False
exp2 = False
## Sultani (without big videos)
if sul:
train_0_path = './training_splits/Sultani/train_names810_sultani_class0.txt'
train_1_path = './training_splits/Sultani/train_names800_sultani_class1.txt'
num_0 = 810
num_1 = 800
Ann_path = './training_splits/Sultani/test_annot310_sultani.txt'
NamesAnn_path = './training_splits/Sultani/test_names310_sultani.txt'
## Experiment 1
if exp1:
train_0_path = './training_splits/Experiment1/train_names750_exp1_class0.txt'
train_1_path = './training_splits/Experiment1/train_names750_exp1_class1_nobig.txt'
num_0 = 750
num_1 = 750
Ann_path = './training_splits/Experiment1/test_annot310_exp1.txt'
NamesAnn_path = './training_splits/Experiment1/test_names310_exp1.txt'
## Experiment 2
if exp2:
train_0_path = './training_splits/Experiment2/train_names750_exp2_class0.txt'
train_1_path = './training_splits/Experiment2/train_names750_exp2_class1_nobig.txt'
num_0 = 750
num_1 = 750
Ann_path = './training_splits/Experiment2/test_annot310_exp2.txt'
NamesAnn_path = './training_splits/Experiment2/test_names310_exp2.txt'