From 1c540d877170f6b9d8f30c3ef4ded6849db8814a Mon Sep 17 00:00:00 2001 From: megemini Date: Tue, 26 Nov 2024 22:16:03 +0800 Subject: [PATCH] [Fix] panns predict.py --- paddlespeech/cls/exps/panns/deploy/predict.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/paddlespeech/cls/exps/panns/deploy/predict.py b/paddlespeech/cls/exps/panns/deploy/predict.py index 1dd0fb531..7873aa536 100644 --- a/paddlespeech/cls/exps/panns/deploy/predict.py +++ b/paddlespeech/cls/exps/panns/deploy/predict.py @@ -15,6 +15,7 @@ import argparse import os import numpy as np +import paddle from paddle import inference from paddle.audio.datasets import ESC50 from paddle.audio.features import LogMelSpectrogram @@ -74,6 +75,11 @@ class Predictor(object): self.batch_size = batch_size model_file = os.path.join(model_dir, "inference.pdmodel") + if not os.path.exists(model_file): + model_file = os.path.join(model_dir, "inference.json") + if not os.path.exists(model_file): + raise ValueError("Inference model file not exists!") + params_file = os.path.join(model_dir, "inference.pdiparams") assert os.path.isfile(model_file) and os.path.isfile(