def main(): path = 'C:/Users/Phi Thien/PycharmProjects/TDT76/' dataset = 'piano_roll_fs5' training_data = dataprep.load_all_dataset('datasets/training/' + dataset, binarize=True) names = dataprep.load_all_dataset_names('datasets/training/' + dataset) test = dataprep.test_piano_roll(training_data[0], 15, fs=5) # dataprep.piano_roll_to_mid_file(training_data[0], 'test1.mid', fs=5) # dataprep.visualize_piano_roll(training_data[0], fs=5) generalist = Generalist([128, 31, 31, 31, 128], training_data) generalist.train_network(250) generalist.save_network('generalist.h5') generalist_result = generalist.gen_music(test.T) dataprep.visualize_piano_roll(generalist_result, fs=5) dataprep.piano_roll_to_mid_file(generalist_result, 'gen_res.mid', fs=5)
def view_pianoroll(self, idx): datp.visualize_piano_roll(self[idx])