Artanic30 / MacCap

AAAI 2024 Accepted Paper Mining Fine-Grained Image-Text Alignment for Zero-Shot Captioning via Text-Only Training
8 stars 1 forks source link

How to get decap_cc3m.pth? #9

Open SongrainZhou opened 3 months ago

SongrainZhou commented 3 months ago

I am very interested In your project, but when i run the train_coco.sh, I find that the code needs decap_cc3m.pth. Can you provide this file or tell me how to get this file? Thank you!!

Artanic30 commented 3 months ago

Hi, thanks for your interest. This file is used to reproduce the DeCap results, which contain a 500k clip text feature sampled from cc3m. You may comment lines from 63 to 66 in models/BaseModel.py so the code can run without error. If you want this file for other usage, I can upload it to Google Drive.

1013990424 commented 3 months ago

I am very interested in this PTH file. Please upload it. Thank you