Fix Input Type Error raised in infer()
Browse files- convert_to_mixed.py +2 -2
convert_to_mixed.py
CHANGED
|
@@ -34,8 +34,8 @@ def load_and_preprocess_image(image_path, size=(224, 224)):
|
|
| 34 |
|
| 35 |
def infer(model_path, input_feed):
|
| 36 |
session = ort.InferenceSession(model_path, providers=PROVIDERS)
|
| 37 |
-
input_name = session.get_inputs()[0].name
|
| 38 |
-
result = session.run(None,
|
| 39 |
return result
|
| 40 |
|
| 41 |
def main(args):
|
|
|
|
| 34 |
|
| 35 |
def infer(model_path, input_feed):
|
| 36 |
session = ort.InferenceSession(model_path, providers=PROVIDERS)
|
| 37 |
+
#input_name = session.get_inputs()[0].name
|
| 38 |
+
result = session.run(None, input_feed)
|
| 39 |
return result
|
| 40 |
|
| 41 |
def main(args):
|