-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathexperiments_sd_dino.txt
executable file
·37 lines (36 loc) · 5.61 KB
/
experiments_sd_dino.txt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
# Install requirements for SD_DINO (uncomment)
# pip uninstall open-clip-torch
# pip install -r requirements_model_sd_dino.txt
# Experiments 1_1 to 1_5:
# Experiment 2_1: SD with captions
python src/dataset_collection/generate_descriptors.py --identifier 2_1 --ignore_duplicates TRUE --disable_timestamp TRUE --model SD_DINO --fuse_dino TRUE --only_dino FALSE --dino_v2 TRUE --load_size 448 --sd_load_size 960 --dataset_path $dataset_dir --descriptor_dir $output_dir --combine_pca FALSE --pca TRUE --stride 7 --allow_text_captions TRUE
# Experiment 1_4: SD + DINOv1
python src/dataset_collection/generate_descriptors.py --identifier 1_4 --ignore_duplicates TRUE --disable_timestamp TRUE --model SD_DINO --fuse_dino TRUE --only_dino FALSE --dino_v2 FALSE --load_size 448 --sd_load_size 960 --dataset_path $dataset_dir --descriptor_dir $output_dir --combine_pca FALSE --pca TRUE --stride 4
# Experiment 1_5: SD + DINOv2
python src/dataset_collection/generate_descriptors.py --identifier 1_5 --ignore_duplicates TRUE --disable_timestamp TRUE --model SD_DINO --fuse_dino TRUE --only_dino FALSE --dino_v2 TRUE --load_size 448 --sd_load_size 960 --dataset_path $dataset_dir --descriptor_dir $output_dir --combine_pca FALSE --pca TRUE --stride 7
# Experiment 1_3: SD ONLY
python src/dataset_collection/generate_descriptors.py --identifier 1_3 --ignore_duplicates TRUE --disable_timestamp TRUE --model SD_DINO --fuse_dino FALSE --only_dino FALSE --dino_v2 FALSE --load_size 448 --sd_load_size 960 --dataset_path $dataset_dir --descriptor_dir $output_dir --combine_pca FALSE --pca TRUE --stride 4
# Experiment 1_1: DINOv1
python src/dataset_collection/generate_descriptors.py --identifier 1_1 --ignore_duplicates TRUE --disable_timestamp TRUE --model SD_DINO --fuse_dino TRUE --only_dino TRUE --dino_v2 FALSE --load_size 448 --dataset_path $dataset_dir --descriptor_dir $output_dir --combine_pca FALSE --pca TRUE --stride 4
# Experiment 1_1_2: DINOv1
python src/dataset_collection/generate_descriptors.py --identifier 1_1_2 --ignore_duplicates TRUE --disable_timestamp TRUE --model SD_DINO --fuse_dino TRUE --only_dino TRUE --dino_v2 FALSE --load_size 448 --dataset_path $dataset_dir --descriptor_dir $output_dir --combine_pca FALSE --pca TRUE --stride 8
# Experiment 1_2: DINOv2
python src/dataset_collection/generate_descriptors.py --identifier 1_2 --ignore_duplicates TRUE --disable_timestamp TRUE --model SD_DINO --fuse_dino TRUE --only_dino TRUE --dino_v2 TRUE --load_size 448 --dataset_path $dataset_dir --descriptor_dir $output_dir --combine_pca FALSE --pca TRUE --stride 7
# Experiment 1_2_2: DINOv2
python src/dataset_collection/generate_descriptors.py --identifier 1_2_2 --ignore_duplicates TRUE --disable_timestamp TRUE --model SD_DINO --fuse_dino TRUE --only_dino TRUE --dino_v2 TRUE --load_size 448 --dataset_path $dataset_dir --descriptor_dir $output_dir --combine_pca FALSE --pca TRUE --stride 7 --layer 9
# Experiment 1_2_3: DINOv2
python src/dataset_collection/generate_descriptors.py --identifier 1_2_3 --ignore_duplicates TRUE --disable_timestamp TRUE --model SD_DINO --fuse_dino TRUE --only_dino TRUE --dino_v2 TRUE --load_size 448 --dataset_path $dataset_dir --descriptor_dir $output_dir --combine_pca FALSE --pca TRUE --stride 7 --layer 5
# Experiment 1_3_2: SD ONLY REPEAT as previous SD experiments used true 960 not lossy rescaling i.e. (size -> 448 -> 960) which is unfair...
# theoretically with this and the DINOv1 and DINOv2 experiments, we can recreate SD + DINOv1 and SD + DINOv2 and explore alpha values...
# python src/dataset_collection/generate_descriptors.py --identifier 1_3_2 --ignore_duplicates TRUE --disable_timestamp TRUE --model SD_DINO --fuse_dino FALSE --only_dino FALSE --dino_v2 FALSE --load_size 448 --sd_load_size 960 --dataset_path $dataset_dir --descriptor_dir $output_dir --combine_pca FALSE --pca TRUE --stride 4
# Experiment 1_3_3: SD raw with s4
python src/dataset_collection/generate_descriptors.py --identifier 1_3_3 --ignore_duplicates TRUE --disable_timestamp TRUE --model SD_DINO --fuse_dino FALSE --only_dino FALSE --dino_v2 TRUE --load_size 448 --sd_load_size 960 --dataset_path $dataset_dir --descriptor_dir $output_dir --pca FALSE --raw TRUE --raw_layer s4
# Experiment 1_3_4: SD raw with s5
python src/dataset_collection/generate_descriptors.py --identifier 1_3_4 --ignore_duplicates TRUE --disable_timestamp TRUE --model SD_DINO --fuse_dino FALSE --only_dino FALSE --dino_v2 TRUE --load_size 448 --sd_load_size 960 --dataset_path $dataset_dir --descriptor_dir $output_dir --pca FALSE --raw TRUE --raw_layer s5
# Experiment 1_3_5: SD with mask
python src/dataset_collection/generate_descriptors.py --identifier 1_3_5 --ignore_duplicates TRUE --disable_timestamp TRUE --model SD_DINO --fuse_dino FALSE --only_dino FALSE --dino_v2 TRUE --load_size 448 --sd_load_size 960 --dataset_path $dataset_dir --descriptor_dir $output_dir --pca FALSE --mask TRUE
# Experiment 2_2: SD with captions
python src/dataset_collection/generate_descriptors.py --identifier 2_2 --ignore_duplicates TRUE --disable_timestamp TRUE --model SD_DINO --fuse_dino TRUE --only_dino FALSE --dino_v2 TRUE --load_size 448 --sd_load_size 960 --dataset_path $dataset_dir --descriptor_dir $output_dir --combine_pca FALSE --pca FALSE --raw TRUE --raw_layer s4 --stride 7 --allow_text_captions TRUE
# Experiment 2_3: SD with captions
python src/dataset_collection/generate_descriptors.py --identifier 2_3 --ignore_duplicates TRUE --disable_timestamp TRUE --model SD_DINO --fuse_dino TRUE --only_dino FALSE --dino_v2 TRUE --load_size 448 --sd_load_size 960 --dataset_path $dataset_dir --descriptor_dir $output_dir --combine_pca FALSE --pca FALSE --raw TRUE --raw_layer s5 --stride 7 --allow_text_captions TRUE
# End of SD_DINO