search

chenxi52 / F-VLM_pytorch

Pytorch version of F-VLM
2 stars 0 forks source link

关于coco_cls.pkl和lvis_cls.pkl的问题 #2

Open pulgawang opened 1 month ago

pulgawang commented 1 month ago

您好，我最近在研读您的代码，在调试代码的时候发现以下问题：

  1. 我根据FVLM demo.py代码中生成了一份coco_embedding，但是发现和您提供的coco_cls.pkl中的数据不一致，生成代码如下：

    coco_embeddings = np.concatenate((text_embeddings, background_embedding), axis=0)  # 在L112后位置增加并保存

    数据对比附在最后

  2. 希望您能提供lvis_cls.pkl文件，在项目中没有找到该文件

谢谢

自己生成的coco_embedding和coco_cls.pkl的对比

chenxi52 commented 1 month ago

我的是自己生成的，我记得使用F-VLM产生的embedding没有自己生成的这个跑的效果好，具体原因不知，应该实验环境什么的都不一样。因为COCO APrare没有赶上他，关于lvis的实验可能尝试了也没超过，具体细节已经忘记了。另外，train_net_stand.py 注释掉的代码里面（L266) 应该可以生成lvis_cls.pkl。祝好

pulgawang commented 1 month ago

谢谢，我试一下