Fix script to handle different size of inputs

P.808 model needs 144000 samples while P.835 model needs 144160 samples
This commit is contained in:
Vishak Gopal 2023-01-09 22:10:28 -08:00 коммит произвёл GitHub
Родитель 76f61fa1e5
Коммит 6412bc5834
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
1 изменённых файлов: 1 добавлений и 1 удалений

Просмотреть файл

@ -74,7 +74,7 @@ class ComputeScore:
continue
input_features = np.array(audio_seg).astype('float32')[np.newaxis,:]
p808_input_features = np.array(self.audio_melspec(audio=audio_seg)).astype('float32')[np.newaxis, :, :]
p808_input_features = np.array(self.audio_melspec(audio=audio_seg[:-160])).astype('float32')[np.newaxis, :, :]
oi = {'input_1': input_features}
p808_oi = {'input_1': p808_input_features}
p808_mos = self.p808_onnx_sess.run(None, p808_oi)[0][0][0]