From 20b2d7a958d09821ea5e30c1cc204b8c6908d042 Mon Sep 17 00:00:00 2001 From: Malar Kannan Date: Mon, 27 Nov 2017 14:08:01 +0530 Subject: [PATCH] updated model data --- .gitignore | 2 ++ speech_data.py | 7 +++---- speech_model.py | 2 +- speech_test.py | 4 ++-- 4 files changed, 8 insertions(+), 7 deletions(-) diff --git a/.gitignore b/.gitignore index d2d0b11..3ce1b01 100644 --- a/.gitignore +++ b/.gitignore @@ -143,3 +143,5 @@ inputs/audio* logs/* models/* *.pkl +temp/* +trained/* diff --git a/speech_data.py b/speech_data.py index 9b3ea54..7eff4fb 100644 --- a/speech_data.py +++ b/speech_data.py @@ -147,7 +147,7 @@ def read_siamese_tfrecords_generator(audio_group='audio',batch_size=32,test_size (n_spec,n_features,n_records) = pickle.load(open(const_file,'rb')) def copy_read_consts(dest_dir): - shutil.copy2(const_file,dest_dir) + shutil.copy2(const_file,dest_dir+'/constants.pkl') return (n_spec,n_features,n_records) # @threadsafe_iter def record_generator(): @@ -259,9 +259,8 @@ if __name__ == '__main__': # create_spectrogram_tfrecords('story_all',sample_count=25) # fix_csv('story_words_test') #fix_csv('audio') - # create_spectrogram_tfrecords('story_words_test',sample_count=100,train_test_ratio=0.1) - print(record_generator_count('outputs/story_phrases.full.train.tfrecords')) - print(record_generator_count('outputs/story_phrases.full.test.tfrecords')) + create_spectrogram_tfrecords('story_words_test',sample_count=10,train_test_ratio=0.1) + #record_generator_count() # create_spectrogram_tfrecords('audio',sample_count=50) # read_siamese_tfrecords_generator('audio') # padd_zeros_siamese_tfrecords('audio') diff --git a/speech_model.py b/speech_model.py index ff2822e..9b9f281 100644 --- a/speech_model.py +++ b/speech_model.py @@ -124,4 +124,4 @@ def train_siamese(audio_group = 'audio'): if __name__ == '__main__': - train_siamese('story_phrases.full') + train_siamese('story_words_test') diff --git a/speech_test.py b/speech_test.py index 26f5a54..4fcd2fc 100644 --- a/speech_test.py +++ b/speech_test.py @@ -29,7 +29,7 @@ def test_with(audio_group): def evaluate_siamese(records_file,audio_group='audio',weights = 'siamese_speech_model-final.h5'): # audio_group='audio';model_file = 'siamese_speech_model-305-epoch-0.20-acc.h5' # records_file = os.path.join('./outputs',eval_group+'.train.tfrecords') - const_file = os.path.join('./models/'+audio_group+'/',audio_group+'.constants') + const_file = os.path.join('./models/'+audio_group+'/','constants.pkl') arch_file='./models/'+audio_group+'/siamese_speech_model_arch.yaml' weight_file='./models/'+audio_group+'/'+weights (n_spec,n_features,n_records) = pickle.load(open(const_file,'rb')) @@ -177,7 +177,7 @@ def visualize_results(audio_group='audio'): if __name__ == '__main__': # evaluate_siamese('./outputs/story_words_test.train.tfrecords',audio_group='story_words.gpu',weights ='siamese_speech_model-58-epoch-0.00-acc.h5') # evaluate_siamese('./outputs/story_words.test.tfrecords',audio_group='story_words',weights ='siamese_speech_model-675-epoch-0.00-acc.h5') - evaluate_siamese('./outputs/story_phrases.test.tfrecords',audio_group='story_phrases',weights ='siamese_speech_model-956-epoch-0.20-acc.h5') + evaluate_siamese('./outputs/story_words_test.train.tfrecords',audio_group='story_words_test.10',weights ='siamese_speech_model-891-epoch-0.02-acc.h5') # play_results('story_words') #inspect_tfrecord('./outputs/story_phrases.test.tfrecords',audio_group='story_phrases') # visualize_results('story_words.gpu')