def main(unused_argv): if FLAGS.out_file is None: FLAGS.out_file = FLAGS.in_file with tf.gfile.Open(FLAGS.in_file, 'r') as f: pianorolls = pickle.load(f) dense_pianorolls = [sparse_pianoroll_to_dense(p, MIN_NOTE, NUM_NOTES)[0] for p in pianorolls['train']] # Concatenate all elements along the time axis. concatenated = np.concatenate(dense_pianorolls, axis=0) mean = np.mean(concatenated, axis=0) pianorolls['train_mean'] = mean # Write out the whole pickle file, including the train mean. pickle.dump(pianorolls, open(FLAGS.out_file, 'wb'))