diff --git a/benchmark/run.py b/benchmark/run.py index 6672eee..2c2d5a3 100644 --- a/benchmark/run.py +++ b/benchmark/run.py @@ -119,6 +119,10 @@ elif args.format == 'onnx': if not os.path.exists(onnx_path): onnx_path = op.save_model(format='onnx') + sess = onnxruntime.InferenceSession(onnx_path, + providers=onnxruntime.get_available_providers()) + + @towhee.register def run_onnx(img): img = img.convert('RGB')