train_config_improved.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": 500,
  "backbone_lr": 1e-05,
  "decoder_lr": 0.0001,
  "weight_decay": 0.0001,
  "scheduler": "cosine",
  "grad_clip": 1.0,
  "focal_alpha": [1.0, 3.0],
  "focal_gamma": 2,
  "dice_weight": 0.5,
  "focal_weight": 1.0,
  "background_weight": 1.0,
  "foreground_weight": 3.0,
  "num_workers": 0,
  "output_dir": "outputs/seaweed_segmentation_improved_epoch500",
  "plot_interval": 10,
  "device": "cuda"
}