Esempio n. 1
0
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'))
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'))