config.json

{
  "format_version": 1,
  "model": {
    "class_names": [
      "background",
      "blue",
      "green",
      "red"
    ],
    "image_size": [
      576,
      768
    ],
    "image_mean": [
      0.485,
      0.456,
      0.406
    ],
    "image_std": [
      0.229,
      0.224,
      0.225
    ],
    "padding_color": [
      124,
      116,
      104
    ],
    "hidden_state_indices": [
      3,
      6,
      9,
      12
    ],
    "dpt_features": 64,
    "dpt_channels": [
      48,
      96,
      192,
      384
    ],
    "wt_kernel_size": 5,
    "wt_levels": 2,
    "wt_type": "db1",
    "backbone": {
      "model_type": "dinov3_vit",
      "architectures": [
        "DINOv3ViTModel"
      ],
      "attention_dropout": 0.0,
      "drop_path_rate": 0.0,
      "hidden_act": "gelu",
      "hidden_size": 384,
      "image_size": 224,
      "initializer_range": 0.02,
      "intermediate_size": 1536,
      "key_bias": false,
      "layer_norm_eps": 1e-05,
      "layerscale_value": 1.0,
      "mlp_bias": true,
      "num_attention_heads": 6,
      "num_channels": 3,
      "num_hidden_layers": 12,
      "num_register_tokens": 4,
      "patch_size": 16,
      "pos_embed_jitter": null,
      "pos_embed_rescale": 2.0,
      "pos_embed_shift": null,
      "proj_bias": true,
      "query_bias": true,
      "rope_theta": 100.0,
      "use_gated_mlp": false,
      "value_bias": true
    }
  },
  "checkpoint": {
    "filename": "model.safetensors",
    "size_bytes": 96876480,
    "sha256": "2fdec883d6a94b7a3478a750f6943bb41b60abcb8316649bd8a050f9bd3e5f4d",
    "training_epoch": 66,
    "best_validation_miou_foreground": 0.40750709018075176,
    "test_miou_foreground": 0.3633164423801378,
    "test_mean_dice_foreground": 0.5295766998594339
  }
}