File size: 547 Bytes
25cae60 |
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 |
model:
arch: vit_small
patch_size: 8
pre_training: dino
found:
bkg_th: 0.3
feats: k
training:
dataset: DUTS-TR
dataset_set: null
# Hyper params
seed: 0
max_iter: 500
nb_epochs: 3
batch_size: 50
lr0: 5e-2
step_lr_size: 50
step_lr_gamma: 0.95
w_bs_loss: 1.5
stop_bkg_loss: 100
# Augmentations
crop_size: 224
scale_range: [0.1, 3.0]
photometric_aug: gaussian_blur
proba_photometric_aug: 0.5
cropping_strategy: random_scale
evaluation:
type: saliency # uod, retrieval
datasets: [DUT-OMRON, ECSSD] |