Hugging Face's logo

Spaces:
ccmusic-database
/
erhu_playing_tech
Running

App Files
erhu_playing_tech
File size: 3,070 Bytes 
14e7761
 
 
d574298
 
14e7761
 
d574298
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
14e7761
 
 
d574298
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
14e7761
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
db5e575
14e7761
d574298
 
14e7761
d574298
14e7761
d574298
14e7761
 
d574298
14e7761
db5e575
 
 
 
 
14e7761
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
 
import os
import torch
import torchvision.transforms as transforms
import huggingface_hub
import modelscope
from PIL import Image

EN_US = os.getenv("LANG") != "zh_CN.UTF-8"

ZH2EN = {
    "上传录音": "Upload a recording",
    "选择模型": "Select a model",
    "状态栏": "Status",
    "音频文件名": "Audio filename",
    "演奏技法识别": "Playing tech recognition",
    "建议录音时长保持在 3s 左右": "It is recommended to keep the recording length around 3s.",
    "引用": "Cite",
    "揉弦": "Rou xian",
    "颤音": "Chan yin",
    "颤弓": "Chan gong",
    "顿弓": "Dun gong",
    "抛弓": "Pao gong",
    "拨弦": "Bo xian",
    "击弓": "Ji gong",
    "连滑音": "Lian hua yin",
    "泛音": "Fan yin",
    "垫弓": "Dian gong",
    "分弓": "Fen gong",
}

MODEL_DIR = (
    huggingface_hub.snapshot_download(
        "ccmusic-database/erhu_playing_tech",
        cache_dir="./__pycache__",
    )
    if EN_US
    else modelscope.snapshot_download(
        "ccmusic-database/erhu_playing_tech",
        cache_dir="./__pycache__",
    )
)


def _L(zh_txt: str):
    return ZH2EN[zh_txt] if EN_US else zh_txt


TRANSLATE = {
    "vibrato": _L("揉弦"),
    "trill": _L("颤音"),
    "tremolo": _L("颤弓"),
    "staccato": _L("顿弓"),
    "ricochet": _L("抛弓"),
    "pizzicato": _L("拨弦"),
    "percussive": _L("击弓"),
    "legato_slide_glissando": _L("连滑音"),
    "harmonic": _L("泛音"),
    "diangong": _L("垫弓"),
    "detache": _L("分弓"),
}
CLASSES = list(TRANSLATE.keys())
TEMP_DIR = "./__pycache__/tmp"
SAMPLE_RATE = 44100


def toCUDA(x):
    if hasattr(x, "cuda"):
        if torch.cuda.is_available():
            return x.cuda()

    return x


def find_wav_files(folder_path=f"{MODEL_DIR}/examples"):
    wav_files = []
    for root, _, files in os.walk(folder_path):
        for file in files:
            if file.endswith(".wav"):
                file_path = os.path.join(root, file)
                wav_files.append(file_path)

    return wav_files


def get_modelist(model_dir=MODEL_DIR, assign_model=""):
    output = []
    for entry in os.listdir(model_dir):
        # 获取完整路径
        full_path = os.path.join(model_dir, entry)
        # 跳过'.git'文件夹
        if entry == ".git" or entry == "examples":
            print(f"跳过 .git 或 examples 文件夹: {full_path}")
            continue

        # 检查条目是文件还是目录
        if os.path.isdir(full_path):
            model = os.path.basename(full_path)
            if assign_model and assign_model.lower() in model:
                output.insert(0, model)
            else:
                output.append(model)

    return output


def embed_img(img_path: str, input_size=224):
    transform = transforms.Compose(
        [
            transforms.Resize([input_size, input_size]),
            transforms.ToTensor(),
            transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)),
        ]
    )
    img = Image.open(img_path).convert("RGB")
    return transform(img).unsqueeze(0)