Closed songbingyue closed 1 year ago
Yes, you can see https://github.com/keyu-tian/SparK/tree/main/pretrain#tutorial-for-pretraining-your-own-dataset.
The only thing needed to do is to replace the function build_dataset_to_pretrain
in line54-75 of pretrain/utils/imagenet.py to yours. This function should return a Dataset object. You may use args like args.data_path
and args.input_size
to help build your dataset. And when runing experiment with main.sh
you can use --data_path=... --input_size=...
to specify them.
Can I replace imagenet1k with another dataset?