Программа распознавания голоса работает спорадически - PullRequest
Новая
       33

Программа распознавания голоса работает спорадически

0 голосов
/ 06 сентября 2018

Я не уверен, почему, но он только находит, конвертирует и транскрибирует все mp3-файлы в каталоге время от времени, но никогда в 100% случаев.Я не уверен почему.Я надеюсь, что я спрашиваю в нужном месте.Моя цель - найти все файлы m4a, а затем преобразовать их в файлы wav, а затем найти все файлы wav и записать их.Программа делает это время от времени, но не все время. 

#!/usr/bin/env python3

    import speech_recognition as sr
    import time
    import subprocess
    import os
    from os import path
    from os import listdir


    # find and convert mp3 files to wav files

    # find all files with an extension, convert to wav files, transcribe to text file then transfer all wav files and mp3 files to finished directory and email transcribed files.

    # find files with mp3 extension

    def list_files1(directory, extension):
        ext = []
        for root, dirs, files in os.walk(directory):
            for file in files:
                if file.endswith(extension):
                    ext.append(file)
        print(ext)
        return ext


    # get directory path
    originalDir = os.getcwd()

    # call function to find files with mp3 extension
    mp3files = list_files1(originalDir, "m4a")

    # os.chdir("/Users/williamjohnson/Dropbox/goodscrapers/publimaison2/")

    # convert all mp3 files to wav files
    for x in mp3files:
        print(x)
        timestr = time.strftime("%Y%m%d-%H%M%S")
        command = "ffmpeg -i " + x + ' ' + timestr + ".wav"
        print(command)
        subprocess.call(command, shell=True)

    # find all converted wav files

    wavfiles = list_files1(originalDir, "wav")

    for y in wavfiles:
        print(y)
        # obtain path to "english.wav" in the same folder as this script
        AUDIO_FILE = path.join(path.dirname(path.realpath(__file__)), y)
        # AUDIO_FILE = path.join(path.dirname(path.realpath(__file__)), "french.aiff")
        # AUDIO_FILE = path.join(path.dirname(path.realpath(__file__)), "chinese.flac")

        # use the audio file as the audio source
        r = sr.Recognizer()
        with sr.AudioFile(AUDIO_FILE) as source:
            audio = r.record(source)  # read the entire audio file

        # recognize speech using Sphinx
        try:
            print("Sphinx thinks you said " + r.recognize_sphinx(audio))
            timestr = time.strftime("%Y%m%d-%H%M%S")
            text_file = open(timestr + ".txt", "a")
            text_file.write(r.recognize_sphinx(audio))
            text_file.close()
        except sr.UnknownValueError:
            print("Sphinx could not understand audio")
        except sr.RequestError as e:
            print("Sphinx error; {0}".format(e))

Редактировать: Я сделал очень глупую ошибку, я назвал все выведенные текстовые файлы с одинаковым именем, чтобы ониперезаписывая, я удостоверился, что дал им уникальное имя, опустившись до миллисекунды в качестве имени и затем добавив случайное число к имени файла для хорошей меры.

1 Ответ

0 голосов
/ 06 сентября 2018

Когда вы используете os.walk(), он возвращает файлы без какого-либо каталога , поэтому вы собираете список файлов, но отбрасываете их имя каталога.

Кажется, ваша оболочка не добавляет никакого значения сверх os.walk(); Я бы в любом случае рефакторинг для преобразования одного файла за раз. Тангенциально, вам также не нужно звонить getcwd, если вам не особенно важен абсолютный путь к текущему каталогу. 

import os

def get_next_file(directory, extension):
    for root, dirs, files in os.walk(directory):
        for file in files:
            if file.endswith(extension):
                yield os.path.join(root, file)

for x in get_next_file('.', 'm4a'):
    print(x)
...