ibrahim313's picture
Upload config.json with huggingface_hub
2840c1d verified
raw
history blame contribute delete
978 Bytes
{
"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
}
}