File size: 751 Bytes
e8ace62 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
import torch
from feature_matcher_utilities import extract_keypoints, feature_matching
from lightglue import LightGlue
path_0 = "/media/alejandro/E45B-3EBD/FOR ALEX/8312297.jpg"
path_1 = "/media/alejandro/E45B-3EBD/FOR ALEX/8315088.jpg"
feats_dict0, h0, w0 = extract_keypoints(path_0, features="superpoint", rotations=[0])
feats_dict1, h1, w1 = extract_keypoints(path_1, features="superpoint", rotations=[1])
device = 'cuda' if torch.cuda.is_available() else 'cpu'
matcher = LightGlue(features='superpoint', depth_confidence=-1, width_confidence=-1, flash=True).eval().to(device)
matches_tensor = feature_matching(feats_dict0, feats_dict1, matcher=matcher, features="superpoint", matcher_type="lightglue")
print(len(matches_tensor)) |