Update run_axmodel.py
Browse files- run_axmodel.py +1 -1
run_axmodel.py
CHANGED
|
@@ -75,7 +75,7 @@ def pad_image(img, padding):
|
|
| 75 |
def run(args):
|
| 76 |
'''onnx inference'''
|
| 77 |
# model
|
| 78 |
-
session = axe.InferenceSession(args.model
|
| 79 |
output_names = [x.name for x in session.get_outputs()]
|
| 80 |
input_name = session.get_inputs()[0].name
|
| 81 |
|
|
|
|
| 75 |
def run(args):
|
| 76 |
'''onnx inference'''
|
| 77 |
# model
|
| 78 |
+
session = axe.InferenceSession(args.model)
|
| 79 |
output_names = [x.name for x in session.get_outputs()]
|
| 80 |
input_name = session.get_inputs()[0].name
|
| 81 |
|