rei-forks/precise-lite-amd64aarch64
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Recursive glob data/random

Browse Source
This commit is contained in:
Matthew D. Scholefield 2018-02-23 14:44:50 -06:00
parent 988a4fe60c
commit 88173fef95
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
precise/scripts/train_incremental.py
View File

@ -1,6 +1,5 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
# Copyright (c) 2017 Mycroft AI Inc. # Copyright (c) 2017 Mycroft AI Inc.
from glob import glob
from os import makedirs from os import makedirs
from os.path import basename, splitext, isfile, join from os.path import basename, splitext, isfile, join
from random import random from random import random
@ -13,7 +12,7 @@ from precise.model import create_model
from precise.network_runner import Listener, KerasRunner from precise.network_runner import Listener, KerasRunner
from precise.params import inject_params from precise.params import inject_params
from precise.train_data import TrainData from precise.train_data import TrainData
from precise.util import load_audio, save_audio from precise.util import load_audio, save_audio, glob_all
usage = ''' usage = '''
Train a model to inhibit activation by Train a model to inhibit activation by
@ -138,7 +137,7 @@ class IncrementalTrainer:
Begin reading through audio files, saving false Begin reading through audio files, saving false
activations and retraining when necessary activations and retraining when necessary
""" """
for fn in glob(self.args.random_data_dir + '/*.wav'): for fn in glob_all(self.args.random_data_dir, '*.wav'):
if fn in self.trained_fns: if fn in self.trained_fns:
print('Skipping ' + fn + '...') print('Skipping ' + fn + '...')
continue continue