lhackel-tub / ConfigILM

A Library for configurable combination of pre-configured and possibly pre-trained Image and Language Models
https://lhackel-tub.github.io/ConfigILM/
MIT License
37 stars 5 forks source link

Option to return patch-names for BENDataSet #45

Closed jakhac closed 1 year ago

jakhac commented 1 year ago

For visualization of image retrieval results using BEN, it is necessary to assign patch-names to samples returned from the BENDataSet.__getitem__() method. A convenient extension to this class would therefore be to (optionally) return the patch-names (key) in addition to image and labels.