train_segmentation_frozen_config.json

{
  "train_image_dir": "data/train/images",
  "train_mask_dir": "data/train/masks",
  "val_image_dir": "data/val/images",
  "val_mask_dir": "data/val/masks",
  "num_classes": 2,
  "backbone_name": "dinov3_vitl16",
  "pretrained": true,
  "backbone_weights": "dinov3_vitl16_pretrain_sat493m-eadcf0ff.pth",
  "use_4channel": true,
  "image_size": 256,
  "batch_size": 8,
  "num_epochs": 50,
  "backbone_lr": 1e-05,
  "decoder_lr": 5e-05,
  "weight_decay": 0.0001,
  "scheduler": "cosine",
  "grad_clip": 0.5,
  "focal_alpha": 0.25,
  "focal_gamma": 2,
  "dice_weight": 0.5,
  "focal_weight": 1,
  "num_workers": 0,
  "output_dir": "outputs/seaweed_segmentation_frozen",
  "plot_interval": 10,
  "device": "cuda",
  "freeze_backbone": true,
  "use_data_augmentation": false
}