[Update] unsqueeze

10 months ago · 2279e6fd73
parent 8d2f17672a
commit 2279e6fd73
1 changed files with 1 additions and 1 deletions
--- a/paddlespeech/cls/exps/panns/deploy/predict.py
+++ b/paddlespeech/cls/exps/panns/deploy/predict.py
@ -58,7 +58,7 @@ def extract_features(files: str, **kwargs):

        feature_extractor = LogMelSpectrogram(sr, **kwargs)
        feat = feature_extractor(paddle.to_tensor(waveforms[i]))
-        feat = paddle.transpose(feat, perm=[1, 0])
+        feat = paddle.transpose(feat, perm=[1, 0]).unsqueeze(0)
        feats.append(feat)

    return np.stack(feats, axis=0)