search

KMnP / vpt

❄️🔥 Visual Prompt Tuning [ECCV 2022] https://arxiv.org/abs/2203.12119
Other
1.04k stars 91 forks source link

How to apply data augmentation for VTAB #18

Closed kai-wen-yang closed 2 years ago

kai-wen-yang commented 2 years ago

Hi, I want to apply some data augmentation using torchvision.transform for VTAB dataset. Does your repo support that? If yes, how to implement it? Thanks a lot!

KMnP commented 2 years ago

Hi, we don't support more augmentation techniques in this repo, following practices of original VTAB paper.

But you can add your own inside this function: https://github.com/KMnP/vpt/blob/main/src/data/datasets/tf_dataset.py#L148