| 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)) |