updated tested pickling
parent
88edcdd239
commit
b3755ad80e
|
|
@ -139,3 +139,4 @@ Temporary Items
|
||||||
|
|
||||||
outputs/*
|
outputs/*
|
||||||
inputs/mnist
|
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)
|
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
|
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 = 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.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):
|
def speech_pairs_data(audio_group='audio'):
|
||||||
audio_samples = pd.read_pickle('spectrogram.pkl')
|
audio_samples = pd.read_pickle('outputs/spectrogram.pkl')
|
||||||
y_data = audio_samples['variant'].apply(lambda x:x=='normal').values
|
y_data = audio_samples['variant'].apply(lambda x:x=='normal').values
|
||||||
max_samples = audio_samples['spectrogram'].apply(lambda x:x.shape[0]).max()
|
max_samples = audio_samples['spectrogram'].apply(lambda x:x.shape[0]).max()
|
||||||
sample_size = audio_samples['spectrogram'][0].shape[1]
|
sample_size = audio_samples['spectrogram'][0].shape[1]
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue