Spaces:
Runtime error
Runtime error
File size: 797 Bytes
890de26 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
# -*- coding:utf-8 -*-
# @FileName :eres2net.py
# @Time :2023/8/12 15:52
# @Author :lovemefan
# @Email :[email protected]
import os
import numpy as np
import onnxruntime
from paraformer.runtime.python.model.sv.campplus import Campplus
from paraformer.runtime.python.utils.singleton import singleton
campplus_class = Campplus().__class__
@singleton
class Eres2net(campplus_class):
def __init__(self, onnx_path=None, threshold=0.5):
self.onnx = onnx_path or os.path.join(
os.path.dirname(os.path.dirname(__file__)), "onnx/campplus.onnx"
)
self.sess = onnxruntime.InferenceSession(self.onnx)
self.output_name = [nd.name for nd in self.sess.get_outputs()]
self.threshhold = threshold
self.memory: np.ndarray = None
|