| 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] |