HestiaSky / IMF-Pytorch

MIT License
34 stars 5 forks source link

about the pkl file #9

Open Jajison opened 1 year ago

Jajison commented 1 year ago

could you provide the img_features.pkl and text_features.pkl?

bbshu0801 commented 11 months ago

You can get from here https://github.com/mniepert/mmkb

image
kplxwb commented 4 months ago

你这是啥啊,这里面都是文本数据集,也没有图片啊,它这个图片特征是从什么地方提取的

bbshu0801 commented 4 months ago

https://github.com/mniepert/mmkb

这个链接里不是提到了嘛?你可以用他们的script自己下载

image
kplxwb commented 4 months ago

好吧,我看看

changzhang777 commented 2 months ago

https://github.com/mniepert/mmkb

这个链接里不是提到了嘛?你可以用他们的script自己下载 image

你好,我尝试在DB15K上复现的时候发现用bert读取entity_description.txt的话会得到12842×768的向量,然后MMKB获取的图像特征向量是12841×4096的向量,这导致 IMF-Pytorch-main/layers/layer.py", line 65, in forward x_mm.append(torch.mul(torch.mul(x_modal1, x_modal2), x_modal3)) RuntimeError: The size of tensor a (12842) must match the size of tensor b (12841) at non-singleton dimension 0 请问你是如何解决这个问题的呢?顺便一提,文中报告的DB15K实体有14777个,这和12842或12841都不符合,这是怎么回事呢?期待你的解答,感谢