-
Notifications
You must be signed in to change notification settings - Fork 22
/
Copy pathmodel_constants.py
85 lines (76 loc) · 3.15 KB
/
model_constants.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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
SCORE_INPUT = 78 #score information only
DROP_OUT = 0.2
TOTAL_OUTPUT = 16
NUM_PRIME_PARAM = 11
NUM_TEMPO_PARAM = 1
VEL_PARAM_IDX = 1
DEV_PARAM_IDX = 2
PEDAL_PARAM_IDX = 3
num_second_param = 0
num_trill_param = 5
num_voice_feed_param = 0 # velocity, onset deviation
num_tempo_info = 0
num_dynamic_info = 0 # distance from marking, dynamics vector 4, mean_piano, forte marking and velocity = 4
is_trill_index_score = -11
is_trill_index_concated = -11 - (NUM_PRIME_PARAM + num_second_param)
MEAS_TEMPO_IDX = 13
BEAT_TEMPO_IDX = 11
# test_piece_list = [('schumann', 'Schumann'),
# ('mozart545-1', 'Mozart'),
# ('chopin_nocturne', 'Chopin'),
# ('chopin_fantasie_impromptu', 'Chopin'),
# ('cho_waltz_69_2', 'Chopin'),
# ('lacampanella', 'Liszt'),
# ('bohemian_rhapsody', 'Liszt')
# ]
test_piece_list = [
('bps_5_1', 'Beethoven'),
('bps_27_1', 'Beethoven'),
('bps_17_1', 'Beethoven'),
('bps_7_2', 'Beethoven'),
('bps_31_2', 'Beethoven'),
('bwv_858_prelude', 'Bach'),
('bwv_891_prelude', 'Bach'),
('bwv_858_fugue', 'Bach'),
('bwv_891_fugue', 'Bach'),
('schubert_piano_sonata_664-1', 'Schubert'),
('haydn_keyboard_31_1', 'Haydn'),
('haydn_keyboard_49_1', 'Haydn'),
('schubert_impromptu', 'Schubert'),
('mozart545-1', 'Mozart'),
# ('mozart_symphony', 'Mozart'),
('liszt_pag', 'Liszt'),
('liszt_5', 'Liszt'),
('liszt_9', 'Liszt'),
('chopin_etude_10_2', 'Chopin'),
('chopin_etude_10_12', 'Chopin'),
('chopin_etude_25_12', 'Chopin'),
('cho_waltz_69_2', 'Chopin'),
('chopin_nocturne', 'Chopin'),
('cho_noc_9_1', 'Chopin'),
('chopin_prelude_1', 'Chopin'),
('chopin_prelude_4', 'Chopin'),
('chopin_prelude_5', 'Chopin'),
('chopin_prelude_6', 'Chopin'),
('chopin_prelude_8', 'Chopin'),
('chopin_prelude_15', 'Chopin'),
('kiss_the_rain', 'Chopin'),
('bohemian_rhapsody', 'Liszt'),
('chopin_fantasie_impromptu', 'Chopin', 180),
('schumann', 'Schumann'),
('chopin_barcarolle', 'Chopin'),
('chopin_scherzo', 'Chopin'),
]
# test_piece_list = [
# # ('dmc_glass', 'Liszt', 120),
# ('dmc_prokofiev', 'Prokofiev', 160),
# # ('dmc_shostakovich', 'Bach', 80),
# ('dmc_sho_fugue', 'Liszt', 160),
# # ('dmc_messiaen', 'Debussy', 72)
# ]
emotion_folder_path = 'test_pieces/emotionNet/'
emotion_key_list = ['OR', 'Anger', 'Enjoy', 'Relax', 'Sad']
emotion_data_path = [('Bach_Prelude_1', 'Bach', 1),
('Clementi_op.36-1_mov3', 'Haydn', 3),
('Kuhlau_op.20-1_mov1', 'Haydn', 2),
]