Please follow the installation instructions in pretrain
.
We prepare ImageNet-1K following the instructions in DINOv2 (there are additional files needed compared to the pre-training stage).
-
Specify the config file with
config-file
in the training scriptrun_probing.sh
. -
Use the
pretrained-weights
parameter to provide the path to your pre-trained model. -
Replace
imagenet_path
in thetrain-dataset
andval-dataset
parameters with the directory where your datasets are located. -
Simply run the training script as follows:
bash run_probing.sh
This part is heavily build upon DINOv2. We gratefully thank the authors for their wonderful works.