diff --git a/src/loftr/utils/fine_matching.py b/src/loftr/utils/fine_matching.py index 6e77aded..3f173f30 100644 --- a/src/loftr/utils/fine_matching.py +++ b/src/loftr/utils/fine_matching.py @@ -40,7 +40,7 @@ def forward(self, feat_f0, feat_f1, data): }) return - feat_f0_picked = feat_f0_picked = feat_f0[:, WW//2, :] + feat_f0_picked = feat_f0[:, WW//2, :] sim_matrix = torch.einsum('mc,mrc->mr', feat_f0_picked, feat_f1) softmax_temp = 1. / C**.5 heatmap = torch.softmax(softmax_temp * sim_matrix, dim=1).view(-1, W, W)