data-generation sam_hq_vit_h.pth

[taomrzhang]

During the data generation preparation process, the command wget https://huggingface.co/spaces/xinyu1205/Tag2Text/resolve/main/tag2text_swin_14m.pth needs to be executed. However, when running python run_with_submitit_generate_caption.py, the parameter passed does not use tag2text_swin_14m.pth, but instead the --sam_hq_checkpoint parameter is assigned sam_hq_vit_h.pth. However, I could not find the download link for sam_hq_vit_h.pth. Could the author provide sam_hq_vit_h.pth, or could you guide me to where I can download it? Or, is it possible to use tag2text_swin_14m.pth as a replacement for sam_hq_vit_h.pth?

[frank-xwang]

Hi, you can find sam_hq_vit_h.pth under https://huggingface.co/lkeab/hq-sam/tree/main

Please let me know if you have more questions.