def window_phon(wav):
result = utils.frame_wav(wav, winsize, winstride, window_type)
if result.size > 0:
return result/0.1 #-mean/std
else:
return None
def get_test_matrix_from_wavs(self, winsize, winstride):
pkl_file = os.path.join(
self.checkpoints_folder,
"test_matrix_full" + str(winsize) + "_" + str(winstride) + ".pkl")
if os.path.isfile(pkl_file):
with open(pkl_file, 'rb') as handle:
return pickle.load(handle)
else:
test_matrix = []
wav_files = glob.glob(
os.path.join(self.dataset_folder, "TEST/*/*/*.wav"))
for f in wav_files:
wav, fr = read_wav(f)
result = u.frame_wav(wav, winsize, winstride)
if result.size > 0:
test_matrix.extend(result)
test_matrix = np.array(test_matrix)
with open(pkl_file, 'wb') as handle:
pickle.dump(test_matrix,
handle,
protocol=pickle.HIGHEST_PROTOCOL)
return test_matrix
def get_test_matrix(self,
winsize,
winstride,
window_type=lambda x: np.ones((x, ))):
pkl_file = os.path.join(
self.checkpoints_folder,
"test_matrix_" + str(winsize) + "_" + str(winstride) + ".pkl")
if os.path.isfile(pkl_file):
with open(pkl_file, 'rb') as handle:
return pickle.load(handle)
else:
test_phons = self.get_test_df()
test_matrix = []
for index, row in test_phons.iterrows():
result = u.frame_wav(row['wav'], winsize, winstride,
window_type)
if result.size > 0:
test_matrix.extend(result)
test_matrix = np.array(test_matrix)
with open(pkl_file, 'wb') as handle:
pickle.dump(test_matrix,
handle,
protocol=pickle.HIGHEST_PROTOCOL)
return test_matrix
def window_phon(wav):
result = u.frame_wav(wav, winsize, winstride, window_type)
if result.size > 0:
return result
else:
return None