search

marshuang80 / gloria

GLoRIA: A Multimodal Global-Local Representation Learning Framework forLabel-efficient Medical Image Recognition
Apache License 2.0
173 stars 29 forks source link

image text Retriver #13

Closed Guardian-in-the-WF closed 1 year ago

Guardian-in-the-WF commented 2 years ago

Great work. Could you please supply an example about how to use the class Retriver? Thanks.

marshuang80 commented 1 year ago

Hi @Guardian-in-the-WF, thank you for the interest in our work! The retriever is exactly the same as our zero-shot learning modules:

https://github.com/marshuang80/gloria/blob/416466af1036294301a872e4da169fefc137a192/gloria/gloria.py#L240

But instead of passing in cls_txt_mapping, you'l pass in the retrieving target.

Hope this helps!