No module named 'retrieval'

BUAADreamer / SPN4CIR

[ACM MM 2024] Improving Composed Image Retrieval via Contrastive Learning with Scaling Positives and Negatives

https://arxiv.org/abs/2404.11317

MIT License

11 stars 0 forks source link

No module named 'retrieval' #1

Closed shijiangming1 closed 2 months ago

shijiangming1 commented 2 months ago

Thank you for your excellent work, I meet a bug, as follows:

File "/data1/sjm/ZS_CIR/SPN4CIR/zscir/data_process.py", line 11, in from retrieval import ImageDataset, extract_image_features ModuleNotFoundError: No module named 'retrieval'

Can you help me?

BUAADreamer commented 2 months ago

Sorry for missing this file, I have uploaded it. You can try to pull the latest code.