HestiaSky / IMF-Pytorch

MIT License
34 stars 5 forks source link

[Errno 2] No such file or directory: 'datasets/DB15K/img_features.pkl' #10

Open lk-uoser opened 1 year ago

lk-uoser commented 1 year ago

I can't find the way to generate this file: 'datasets/DB15K/img_features.pkl'; 'datasets/DB15K/text_features.pkl' where is the image feature extraction step? I'd appreciate it if you could answer that for me.

lk-uoser commented 1 year ago

“text_features is the pooled_output of entity descriptions of BERT_base_uncased.” About the answer, I in the link https://huggingface.co/bert-base-uncased? text=Paris+is+the+%5BMASK%5D+of+France The generating method is not found. How to get the pooled_output of entity descriptions of BERT_base_uncased?

bbshu0801 commented 11 months ago

You can refer here https://github.com/google-research/bert

I think they used this model.

image