Skip to content

Commit

Permalink
updated weights
Browse files Browse the repository at this point in the history
  • Loading branch information
Parskatt committed Apr 16, 2024
1 parent ab1d506 commit 55cb056
Showing 1 changed file with 7 additions and 4 deletions.
11 changes: 7 additions & 4 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,8 @@
</p>

## 🆕 v2 Detector
We have updated the trainng recipe for the detector. Weights and updated training code will be posted shortly.
We have updated the training recipe for the detector.
Weights are available here: https://github.com/Parskatt/DeDoDe/releases/download/v2/dedode_detector_L_v2.pth

## 🆕 Kornia Integration
DeDoDe is in kornia,
Expand All @@ -39,11 +40,13 @@ Below we show how DeDoDe can be run, you can also check out the [demos](demo)
from DeDoDe import dedode_detector_L, dedode_descriptor_B, dedode_descriptor_G
from DeDoDe.matchers.dual_softmax_matcher import DualSoftMaxMatcher

detector = dedode_detector_L(weights = torch.load("dedode_detector_L.pth"))
# You can either provide weights manually, or not provide any. If none
# are provided we automatically download them. None: We now use v2 detector weights by default.
detector = dedode_detector_L(weights = None)
# Choose either a smaller descriptor,
descriptor = dedode_descriptor_B(weights = torch.load("dedode_descriptor_B.pth"))
descriptor = dedode_descriptor_B(weights = None)
# Or a larger one
descriptor = dedode_descriptor_G(weights = torch.load("dedode_descriptor_G.pth"),
descriptor = dedode_descriptor_G(weights = None,
dinov2_weights = None) # You can manually load dinov2 weights, or we'll pull from facebook

matcher = DualSoftMaxMatcher()
Expand Down

0 comments on commit 55cb056

Please sign in to comment.