ylingfeng / FGVP

Official Codes for Fine-Grained Visual Prompting, NeurIPS 2023
27 stars 2 forks source link

Can this code enable the model to be trained on my own data? #3

Open bujiebuhuo opened 1 month ago

bujiebuhuo commented 1 month ago

This paper is great. I want to train this model on my own dataset. Can this code be used to train the model?

ylingfeng commented 1 month ago

This paper aims at zero-shot inference with the original capability of VLMs, which do not need to be trained to be applied to third-party datasets. However, if you pursue further fitting on your dataset, you can try training with a VLM adapter and prompt learning, like FGVP equipped with CoOp.