Add end-to-end version of MFA FastSpeech2, test=tts

pull/2693/head
WongLaw 3 years ago
parent 1c19a4a32c
commit 2be3448b4c

@ -11,7 +11,6 @@
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import argparse
import re
import paddle
@ -24,6 +23,8 @@ from paddlespeech.text.models.ernie_linear import ErnieLinear
DefinedClassifier = {
'ErnieLinear': ErnieLinear,
}
class Rhy_predictor():
def __init__(self, model_path, config_path, punc_path):
with open(config_path) as f:
@ -80,15 +81,15 @@ class Rhy_predictor():
def make_rhy_dict(self):
self.rhy_dict = {}
for i, p in enumerate(self.punc_list[1:]):
self.rhy_dict[p] = 'sp'+str(i+1)
self.rhy_dict[p] = 'sp' + str(i + 1)
def pinyin_align(self, pinyins, rhy_pre):
final_py = []
j=0
j = 0
for i in range(len(rhy_pre)):
if rhy_pre[i] in self.rhy_dict:
final_py.append(self.rhy_dict[rhy_pre[i]])
else:
final_py.append(pinyins[j])
j+=1
j += 1
return final_py
Loading…
Cancel
Save