File size: 978 Bytes
2840c1d |
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 |
{
"model_type": "ducknet",
"architecture": "DuckNet-Simplified",
"task": "image-segmentation",
"framework": "pytorch",
"license": "apache-2.0",
"tags": [
"medical-imaging",
"polyp-detection",
"colonoscopy",
"segmentation",
"pytorch",
"ducknet",
"kvasir-seg"
],
"model_config": {
"img_size": [
256,
256
],
"input_channels": 3,
"num_classes": 3,
"architecture_type": "U-Net with Duck-inspired blocks"
},
"training_config": {
"dataset": "Kvasir-SEG",
"epochs": 50,
"batch_size": 8,
"optimizer": "Adam",
"learning_rate": 0.001,
"loss_function": "Jaccard",
"validation_dice": 0.9288,
"validation_jaccard": 0.4892
},
"preprocessing": {
"resize": [
256,
256
],
"normalize": {
"mean": [
0.485,
0.456,
0.406
],
"std": [
0.229,
0.224,
0.225
]
},
"threshold": 0.5
}
} |