Fix database files not loading properly

pull/10/head
Matthew D. Scholefield 2018-03-21 13:07:35 -05:00
parent dcad96fc4e
commit 1ee402fd54
1 changed files with 2 additions and 1 deletions

View File

@ -78,12 +78,13 @@ class TrainData:
if not line:
continue
file, tag = line.split('\t')
db_files[tag.strip()].append(file.strip())
db_files[tag.strip()].append(join(db_folder, file.strip() + '.wav'))
train_files, test_files = ([], []), ([], [])
for label, rows in enumerate([db_files['wake-word'], db_files['not-wake-word']]):
for fn in rows:
if not isfile(fn):
print('Missing file:', fn)
continue
if fn not in train_groups:
train_groups[fn] = (