init

text/symbols.py

@@ -0,0 +1,51 @@
+punctuation = ['!', '?', '…', ",", ".", "'", '-']
+pu_symbols = punctuation + ["SP", "UNK"]
+pad = '_'
+
+# chinese
+zh_symbols = ['E', 'En', 'a', 'ai', 'an', 'ang', 'ao', 'b', 'c', 'ch', 'd', 'e', 'ei', 'en', 'eng', 'er', 'f', 'g', 'h',
+       'i', 'i0', 'ia', 'ian', 'iang', 'iao', 'ie', 'in', 'ing', 'iong', 'ir', 'iu', 'j', 'k', 'l', 'm', 'n', 'o',
+       'ong',
+       'ou', 'p', 'q', 'r', 's', 'sh', 't', 'u', 'ua', 'uai', 'uan', 'uang', 'ui', 'un', 'uo', 'v', 'van', 've', 'vn',
+       'w', 'x', 'y', 'z', 'zh',
+        "AA", "EE", "OO"]
+num_zh_tones = 6
+
+# japanese
+ja_symbols = ['I', 'N', 'U', 'a', 'b', 'by', 'ch', 'cl', 'd', 'dy', 'e', 'f', 'g', 'gy', 'h', 'hy', 'i', 'j', 'k', 'ky',
+              'm', 'my', 'n', 'ny', 'o', 'p', 'py', 'r', 'ry', 's', 'sh', 't', 'ts', 'u', 'V', 'w', 'y', 'z']
+num_ja_tones = 1
+
+# English
+en_symbols = ['aa', 'ae', 'ah', 'ao', 'aw', 'ay', 'b', 'ch', 'd', 'dh', 'eh', 'er', 'ey', 'f', 'g', 'hh', 'ih', 'iy',
+              'jh', 'k', 'l', 'm', 'n', 'ng', 'ow', 'oy', 'p', 'r', 's',
+              'sh', 't', 'th', 'uh', 'uw', 'V', 'w', 'y', 'z', 'zh']
+num_en_tones = 4
+
+# combine all symbols
+normal_symbols = sorted(set(zh_symbols + ja_symbols + en_symbols))
+symbols = [pad] + normal_symbols + pu_symbols
+sil_phonemes_ids = [symbols.index(i) for i in pu_symbols]
+
+# combine all tones
+num_tones = num_zh_tones + num_ja_tones + num_en_tones
+
+# language maps
+language_id_map = {
+    'ZH': 0,
+    "JA": 1,
+    "EN": 2
+}
+num_languages = len(language_id_map.keys())
+
+language_tone_start_map = {
+    'ZH': 0,
+    "JA": num_zh_tones,
+    "EN": num_zh_tones + num_ja_tones
+}
+
+if __name__ == '__main__':
+    a = set(zh_symbols)
+    b = set(en_symbols)
+    print(sorted(a&b))
+