Pad with sample instead of zeros in inference

israelg99 · Mar 2, 2017 · 353e151 · 353e151
1 parent d7bd881
commit 353e151
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/audio_inference.py b/audio_inference.py
@@ -16,9 +16,11 @@
 from eva.util.mutil import sparse_labels
 from eva.util.autil import to_pcm8
 
+#%% Data.
+RATE, DATA = scipy.io.wavfile.read('./data/undertale/undertale_001_once_upon_a_time.comp.wav')
+
 #%% Generation Config.
 print('Preparing generation.')
-RATE = 4000
 UNITS = 356415
 
 #%% Model Config.
@@ -64,8 +66,14 @@ def save_gracefully(signal, frame):
 
 BATCH_SIZE = 1
 
-samples = np.zeros(shape=(BATCH_SIZE, UNITS, BINS))
-audio = np.zeros(shape=(BATCH_SIZE, UNITS))
+samples = np.zeros(shape=(1, UNITS, BINS))
+samples[0, list(range(LENGTH-1)), DATA[:LENGTH-1].astype(int)]
+samples = np.repeat(samples, BATCH_SIZE, axis=0)
+
+audio = np.zeros(shape=(1, UNITS))
+audio[0, :LENGTH-1] = DATA[:LENGTH-1]
+audio = np.repeat(audio, BATCH_SIZE, axis=0)
+
 for i in tqdm(range(UNITS+LENGTH-1)):
     if i >= UNITS-1:
         break