How to train PromptPAR using our own dataset
Closed this issue · 5 comments
VisImage commented
Thank you for sharing the code, but how to train PromptPAR using our own dataset?
1125178969 commented
You can refer to the code in /dataset/prepocess to generate pkl files of your own dataset for training. Just include the image path and onehot labels
VisImage commented
Thanks for reply. There are more than 10 files in the folder. Not clear which file (or files) I need to read to get our own dataset?
1125178969 commented
You can refer to pa100k.py because he is simple enough, you just need to modify attr_words, pa100k_data (annotation file) to your own dataset.
VisImage commented
1125178969 commented
you can find it in PromptPAR