ETH-CVG

/

lightglue_superpoint

like 47

Follow

ETH Zurich - Computer Vision and Geometry Lab 33

Hi,

First of all, thanks for providing this model!

I think there is a small problem with the example. The example currently says one should do:

from transformers import AutoImageProcessor, AutoModel
processor = AutoImageProcessor.from_pretrained("stevenbucaille/lightglue_superpoint")
model = AutoModel.from_pretrained("stevenbucaille/lightglue_superpoint")

However, this does not work. I think the correct call should be:

from transformers import AutoImageProcessor, AutoModel
processor = AutoImageProcessor.from_pretrained("ETH-CVG/lightglue_superpoint")
model = AutoModel.from_pretrained("ETH-CVG/lightglue_superpoint")

This is also what is provided when one clicks on "Use this model".

It's no big deal but it would be great if the example worked out of the box.