Open Zumbalamambo opened 3 years ago
I'm using the following code to estimate the keypoints and matches using onnx
import json import onnxruntime import numpy as np import cv2 path = "output/rgb.png" img = cv2.imread(path) img = cv2.resize(img, dsize=(640, 480), interpolation=cv2.INTER_AREA) img.resize((1, 1, 640, 480)) data = json.dumps({'data': img.tolist()}) data = np.array(json.loads(data)['data']).astype('float32') session = onnxruntime.InferenceSession("output/superpoint_640x480.onnx", None) input_name = session.get_inputs()[0].name output_name = session.get_outputs()[0].name print(input_name) print(output_name) result = session.run([output_name], {input_name: data}) print(result)
How do I interpret the result? or is it the proper way of doing it?
can you get a right result.I tried to make the whole spnet to onnx before,while got an error when inferenced with ort in c++
I'm using the following code to estimate the keypoints and matches using onnx
How do I interpret the result? or is it the proper way of doing it?