tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Change the utils.download_emo_models (#199)

Browse Source 
* Change the utils.download_emo_models

Change utils.download_emo_models(config.mirror, model_name, REPO_ID) to utils.download_emo_models(config.mirror, REPO_ID, model_name)

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
Thymustw
2023-11-27 16:49:00 +08:00
committed by GitHub
parent 2c528ce130
commit badb12585a
1 changed files with 162 additions and 161 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
emo_gen.py
View File

@@ -6,6 +6,7 @@ import librosa
import numpy as np
import torch
import torch.nn as nn
from torch.utils.data import Dataset
from torch.utils.data import DataLoader, Dataset
from tqdm import tqdm
from transformers import Wav2Vec2Processor
@@ -108,7 +109,7 @@ def get_emo(path):
wav, sr = librosa.load(path, 16000)
device = config.bert_gen_config.device
return process_func(
np.expand_dims(wav, 0).astype(np.float64),
np.expand_dims(wav, 0).astype(np.float),
sr,
model,
processor,
@@ -134,7 +135,7 @@ if __name__ == "__main__":
model_name = "./emotional/wav2vec2-large-robust-12-ft-emotion-msp-dim"
REPO_ID = "audeering/wav2vec2-large-robust-12-ft-emotion-msp-dim"
if not Path(model_name).joinpath("pytorch_model.bin").exists():
utils.download_emo_models(config.mirror, model_name, REPO_ID)
utils.download_emo_models(config.mirror, REPO_ID, model_name)
processor = Wav2Vec2Processor.from_pretrained(model_name)
model = EmotionModel.from_pretrained(model_name).to(device)