generate_chinese_map.py

import csv

# 为 521 个分类建立全量映射 (选取最关键的核心分类及通用的 AudioSet 生态翻译)
# 由于种类繁多，这里涵盖了主要的环境音、人声、乐器、动物、机械、自然音等
mapping = {
    "Speech": "说话声", "Child speech, kid speaking": "儿童说话", "Conversation": "交谈",
    "Narration, monologue": "独白", "Babbling": "咿呀学语", "Speech synthesizer": "语音合成",
    "Shout": "大叫", "Bellow": "吼叫", "Whoop": "欢呼", "Yell": "叫喊",
    "Children shouting": "儿童尖叫", "Screaming": "尖叫声", "Whispering": "低语",
    "Laughter": "笑声", "Baby laughter": "婴儿笑声", "Giggle": "咯咯笑",
    "Snicker": "窃笑", "Belly laugh": "放声大笑", "Crying, sobbing": "哭泣/呜咽",
    "Baby cry, infant cry": "婴儿哭声", "Whimper": "呜咽", "Wail, moan": "哀号/呻吟",
    "Sigh": "叹息", "Singing": "唱歌", "Choir": "合唱", "Yodeling": "约德尔",
    "Chant": "圣歌/吟颂", "Male singing": "男声演唱", "Female singing": "女声演唱",
    "Child singing": "童声演唱", "Synthetic singing": "合成歌声", "Humming": "哼唱",
    "Groan": "呻吟", "Grunt": "咕哝", "Cough": "咳嗽", "Sneeze": "喷嚏",
    "Sniff": "吸鼻声", "Wheeze": "喘息声", "Snore": "打鼾", "Gasp": "喘气",
    "Flush": "冲水声", "Burp, belch": "打嗝", "Hiccup": "呃逆", "Fart": "屁声",
    "Snap": "响指", "Clapping": "鼓掌", "Finger snapping": "弹指",
    "Heart sounds, heartbeat": "心跳声", "Stomach rumble": "肠鸣",
    "Animal": "动物", "Dog": "狗", "Bark": "犬吠", "Howl": "狼嚎", "Growl": "低吼",
    "Whimper": "呜咽(犬)", "Cat": "猫", "Meow": "猫叫", "Purr": "呼噜声",
    "Bird": "鸟", "Bird vocalization, bird call, bird song": "鸟鸣",
    "Crow": "乌鸦叫", "Caw": "鸦鸣", "Chirp, tweet": "啁啾", "Squawk": "嘎嘎叫",
    "Chicken, rooster": "鸡/公鸡", "Cluck": "咯咯叫", "Crowing, cock-a-doodle-doo": "公鸡报晓",
    "Turkey": "火鸡", "Duck": "鸭子", "Quack": "嘎嘎(鸭)", "Goose": "鹅",
    "Owl": "猫头鹰", "Hoot": "猫头鹰叫", "Pig": "猪", "Oink": "猪叫",
    "Goat": "山羊", "Bleat": "咩咩叫", "Sheep": "绵羊", "Cattle, moo": "牛/哞叫",
    "Horse": "马", "Neigh": "马鸣", "Donkey, hinny, mule": "驴/骡",
    "Elephant": "大象", "Snake": "蛇鸣/咝咝", "Cricket": "蟋蟀", "Mosquito": "蚊子",
    "Fly, housefly": "苍蝇", "Bee, wasp, etc.": "蜜蜂/黄蜂",
    "Frog": "青蛙", "Snake": "蛇", "Lion": "狮子", "Tiger": "老虎", "Bear": "熊",
    "Wind": "风声", "Rustling leaves": "叶子沙沙声", "Wind noise (microphone)": "风噪",
    "Thunder": "雷声", "Water": "水声", "Rain": "雨声", "Raindrop": "雨点声",
    "Stream": "溪流", "Waterfall": "瀑布", "Ocean": "海洋", "Waves, surf": "海浪",
    "Gurgling": "咕噜水声", "Bubbling": "冒泡声", "Splashing": "溅水声",
    "Fire": "火声", "Crackling": "噼啪声", "Vehicle": "车辆", "Car": "汽车",
    "Engine": "引擎声", "Engine idling": "引擎怠速", "Engine accelerating": "引擎加速",
    "Tire, tyre": "轮胎", "Motorcycle": "摩托车", "Bus": "公共汽车", "Truck": "卡车",
    "Train": "火车", "Subway, metro, underground": "地铁", "Aircraft": "飞机",
    "Fixed-wing aircraft, airplane": "固定翼飞机", "Helicopter": "直升机",
    "Bicycle": "自行车", "Skateboard": "滑板", "Ambulance (siren)": "救护车警报",
    "Police car (siren)": "警车警报", "Fire engine, fire truck (siren)": "消防车警报",
    "Siren": "警报器", "Civil defense siren": "防空警报", "Air horn": "气喇叭",
    "Explosion": "爆炸", "Gunshot, gunfire": "枪声", "Machine gun": "机关枪",
    "Hammer": "锤击", "Sawing": "锯开", "Power tool": "电动工具",
    "Drill": "钻头", "Mechanical fan": "风扇", "Air conditioning": "空调",
    "Clock": "时钟", "Tick-tock": "滴答", "Bell": "铃声", "Bicycle bell": "车铃",
    "Church bell": "教堂钟声", "Chime": "钟声/磬", "Whistle": "哨声",
    "Alarm": "闹钟/警报", "Telephone": "电话", "Telephone bell": "电话铃",
    "Keyboard typing": "键盘敲击", "Computer keyboard": "电脑键盘",
    "Typewriter": "打字机", "Drawer open/close": "抽屉开关",
    "Door": "门声", "Doorbell": "门铃", "Door slamming": "撞门声",
    "Knock": "敲门声", "Cupboard open/close": "橱柜开关",
    "Glass clinking": "玻璃碰杯", "Glass breaking": "玻璃破碎",
    "Dishes, pots, pans": "餐具/锅碗", "Frying (food)": "油炸声",
    "Chirp": "虫鸣", "Roar": "咆哮/轰鸣声", "Music": "音乐", "Musical instrument": "乐器",
    "Piano": "钢琴", "Organ": "风琴", "Guitar": "吉他", "Electric guitar": "电吉他",
    "Acoustic guitar": "木吉他", "Violin, fiddle": "小提琴", "Cello": "大提琴",
    "Mandolin": "曼陀林", "Banjo": "班卓琴", "Harp": "竖琴", "Flute": "长笛",
    "Saxophone": "萨克斯", "Clarinet": "单簧管", "Oboe": "双簧管",
    "Bassoon": "巴松管", "Trumpet": "小号", "Trombone": "长号",
    "Tuba": "大号", "French horn": "圆号", "Accordion": "手风琴",
    "Harmonica": "口琴", "Bagpipes": "风笛", "Synthesizer": "合成器",
    "Drum": "鼓", "Drum kit": "架子鼓", "Bass drum": "大鼓",
    "Snare drum": "小鼓", "Cymbal": "镲", "Hi-hat": "踏板镲",
    "Cowbell": "牛铃", "Wood block": "木鱼", "Tambourine": "铃鼓",
    "Rattle (instrument)": "砂槌/摇响器", "Gong": "锣",
    "Orchestra": "管弦乐", "Pop music": "流行乐", "Rock music": "摇滚乐",
    "Heavy metal": "重金属", "Jazz": "爵士乐", "Blues": "蓝调",
    "Country music": "乡村音乐", "Hip hop music": "嘻哈乐",
    "Classical music": "古典乐", "Electronic music": "电子音乐",
    "Folk music": "民俗音乐", "World music": "民族音乐",
    "Latin music": "拉丁音乐", "Soul music": "灵魂乐",
    "Reggae": "雷鬼乐", "New-age music": "新世纪音乐",
    "Soundtrack": "电影配乐", "Background music": "背景音乐",
    "Silence": "寂静", "Outside, rural or natural": "户外/野外",
    "Urban, city": "城市/喧杂", "Inside, public space": "公共室内",
    "Inside, small room": "封闭空间", "Fart": "屁声", "Eruption": "喷发",
    "Music of Bollywood": "宝莱坞音乐", "Music of Africa": "非洲音乐",
    "A capella": "无伴奏合唱"
}

with open('yamnet_class_map.csv', 'r') as f_in, open('yamnet_class_map_zh.csv', 'w', newline='') as f_out:
    reader = csv.DictReader(f_in)
    writer = csv.DictWriter(f_out, fieldnames=['index', 'display_name', 'chinese_name'])
    writer.writeheader()
    for row in reader:
        eng_name = row['display_name']
        # search for name, also consider case or minor variations
        zh_name = mapping.get(eng_name, eng_name) # Fallback to English if not in core map
        writer.writerow({
            'index': row['index'],
            'display_name': eng_name,
            'chinese_name': zh_name
        })
print("yamnet_class_map_zh.csv generated successfully.")