CervicalSeg / config.json
PlanetSMH's picture
Replace colour model with binary lesion model v0.2.0
ad07800 verified
Raw
History Blame Contribute Delete
2.03 kB
{
"format_version": 2,
"model": {
"class_names": [
"background",
"lesion"
],
"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": 96875968,
"sha256": "2322f48bf36f0fdf77f090c5621c8d31aa6892b51000d40e42f7f43b23c29b98",
"training_epoch": 51,
"checkpoint_selection": "final_epoch",
"training_splits": [
"train",
"val",
"test"
],
"training_sample_counts": {
"train": 1028,
"val": 221,
"test": 216
},
"training_samples_total": 1465,
"final_training_loss": 0.08957859227885158,
"development_validation_foreground_iou_epoch_51": 0.5847087461680525,
"development_validation_foreground_dice_epoch_51": 0.7379384351628312
}
}