我正在短时间内运行简单的 wav2vec2 代码,没有噪音:
#processor = Wav2Vec2Processor.from_pretrained("facebook/wav2vec2-base-960h")
model = Wav2Vec2ForCTC.from_pretrained("facebook/wav2vec2-base-960h")
FILE_NAME = "tutorial-assets/Lab41-SRI-VOiCES-src-sp0307-ch127535-sg0042.wav"
SPEECH_FILE = download_asset(FILE_NAME)
speech, sr = librosa.load(SPEECH_FILE, sr=16000)
speech = torch.tensor(speech)
speech = speech.reshape(1, -1)
logits = model(speech).logits
predicted_ids = torch.argmax(logits, dim=-1)
transcription = processor.decode(predicted_ids[0])
transcription
结果:
'I HAD THAT CURIOSITY BESIDE ME AT THIS MOMENT'
processor
.processor
所以: