File size: 1,760 Bytes
c6dfc69 | 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 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 | # @package _global_
# Video transforms
train_transforms:
- _target_: dataloader.sam2_dataset.transforms.ComposeAPI
transforms:
- _target_: dataloader.sam2_dataset.transforms.RandomHorizontalFlip
consistent_transform: True
- _target_: dataloader.sam2_dataset.transforms.RandomAffine
degrees: 25
shear: 20
image_interpolation: bilinear
consistent_transform: True
- _target_: dataloader.sam2_dataset.transforms.RandomResizeAPI
sizes: 1024
square: true
consistent_transform: True
- _target_: dataloader.sam2_dataset.transforms.ColorJitter
consistent_transform: True
brightness: 0.1
contrast: 0.03
saturation: 0.03
hue: null
- _target_: dataloader.sam2_dataset.transforms.RandomGrayscale
p: 0.05
consistent_transform: True
- _target_: dataloader.sam2_dataset.transforms.ColorJitter
consistent_transform: False
brightness: 0.1
contrast: 0.05
saturation: 0.05
hue: null
- _target_: dataloader.sam2_dataset.transforms.ToTensorAPI
- _target_: dataloader.sam2_dataset.transforms.NormalizeAPI
mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225]
loss:
all:
_target_: loss.training.sam2_training_loss.MultiStepMultiMasksAndIous
weight_dict:
loss_mask: 20
loss_dice: 1
loss_iou: 1
loss_class: 1
supervise_all_iou: true
iou_use_l1_loss: true
pred_obj_scores: true
focal_gamma_obj_score: 0.0
focal_alpha_obj_score: -1.0
gpu_num: 4.
contrastive_learning:
temperature: 0.10
ignore_idx: 255
ood_idx: 254
max_views: 512
proj_dim: 512
sample_limits: 64
total_limits: 15240
|