updated tested pickling
parent
88edcdd239
commit
b3755ad80e
|
|
@ -139,3 +139,4 @@ Temporary Items
|
|||
|
||||
outputs/*
|
||||
inputs/mnist
|
||||
inputs/audio*
|
||||
|
|
|
|||
|
|
@ -56,13 +56,14 @@ def sunflower_pairs_data():
|
|||
te_pairs = np.array([x_pos_test,x_neg_test]).reshape(x_pos_test.shape[0],2,max_samples,sample_size)
|
||||
return tr_pairs,te_pairs,tr_y,te_y
|
||||
|
||||
def create_spectrogram_data():
|
||||
def create_spectrogram_data(audio_group='audio'):
|
||||
audio_samples = pd.read_csv('./outputs/'+audio_group+'.csv',names=['word','voice','rate','variant','file'])
|
||||
# audio_samples = audio_samples.loc[audio_samples['word'] == 'sunflowers'].reset_index(drop=True)
|
||||
audio_samples.loc[:,'spectrogram'] = audio_samples.loc[:,'file'].apply(lambda x:'outputs/'+audio_group+'/'+x).apply(generate_aiff_spectrogram)
|
||||
audio_samples.to_pickle('spectrogram.pkl')
|
||||
audio_samples.to_pickle('outputs/spectrogram.pkl')
|
||||
|
||||
def speech_pairs_data(audio_group):
|
||||
audio_samples = pd.read_pickle('spectrogram.pkl')
|
||||
def speech_pairs_data(audio_group='audio'):
|
||||
audio_samples = pd.read_pickle('outputs/spectrogram.pkl')
|
||||
y_data = audio_samples['variant'].apply(lambda x:x=='normal').values
|
||||
max_samples = audio_samples['spectrogram'].apply(lambda x:x.shape[0]).max()
|
||||
sample_size = audio_samples['spectrogram'][0].shape[1]
|
||||
|
|
|
|||
Loading…
Reference in New Issue