| | |
|---|
| | | input_content: List[Union[np.ndarray, np.ndarray]]) -> np.ndarray: |
|---|
| | | input_dict = dict(zip(self.get_input_names(), input_content)) |
|---|
| | | try: |
|---|
| | | return self.session.run(None, input_dict) |
|---|
| | | return self.session.run(self.get_output_names(), input_dict) |
|---|
| | | except Exception as e: |
|---|
| | | raise ONNXRuntimeError('ONNXRuntime inferece failed.') from e |
|---|
| | | |
|---|
| | |
|---|
| | | |
|---|
| | | |
|---|
| | | @functools.lru_cache() |
|---|
| | | def get_logger(name='rapdi_paraformer'): |
|---|
| | | def get_logger(name='funasr_onnx'): |
|---|
| | | """Initialize and get a logger by name. |
|---|
| | | If the logger has not been initialized, this method will initialize the |
|---|
| | | logger by adding one or two handlers, otherwise the initialized logger will |
|---|
