-
Notifications
You must be signed in to change notification settings - Fork 420
Open
Description
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?
Metadata
Metadata
Assignees
Labels
No labels