Create config.json
Browse files- config.json +17 -0
config.json
ADDED
|
@@ -0,0 +1,17 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"model_type": "unet++",
|
| 3 |
+
"encoder_name": "efficientnet-b4",
|
| 4 |
+
"encoder_weights": "imagenet",
|
| 5 |
+
"in_channels": 3,
|
| 6 |
+
"classes": 1,
|
| 7 |
+
"input_size": [512, 512],
|
| 8 |
+
"architecture": "UnetPlusPlus",
|
| 9 |
+
"library": "segmentation-models-pytorch",
|
| 10 |
+
"metrics": {
|
| 11 |
+
"iou": 0.8558,
|
| 12 |
+
"dice": 0.9223
|
| 13 |
+
},
|
| 14 |
+
"author": "Ruthvik Bandari",
|
| 15 |
+
"license": "cc-by-nc-4.0",
|
| 16 |
+
"github": "https://github.com/Ruthvik-Bandari/DiaFoot.AI"
|
| 17 |
+
}
|