Upload unet3plus+efficientnet model, code, and model card
Browse files- README.md +2 -1
- config.json +7 -6
README.md
CHANGED
|
@@ -56,6 +56,7 @@ Evaluated on the fixed 53-image test partition of Kvasir-SEG (50 % of the origin
|
|
| 56 |
| Precision | 0.9474 |
|
| 57 |
| Recall | 0.9005 |
|
| 58 |
| Accuracy | 0.9745 |
|
|
|
|
| 59 |
|
| 60 |
## Comparison with Sweep Models
|
| 61 |
|
|
@@ -95,7 +96,7 @@ The best trial (#32) achieved `eval_loss = 0.0537` (target: < 0.10 ✓).
|
|
| 95 |
- **Optimiser:** AdamW
|
| 96 |
- **Epochs:** 50 (sweep) + 5 (HPO final retrain)
|
| 97 |
- **FP16:** enabled
|
| 98 |
-
- **Dataset:** Kvasir-SEG augmented (
|
| 99 |
- **Augmentation:** random H/V flips, ±30° rotation, brightness/contrast/saturation ±20 %
|
| 100 |
|
| 101 |
## How to Use
|
|
|
|
| 56 |
| Precision | 0.9474 |
|
| 57 |
| Recall | 0.9005 |
|
| 58 |
| Accuracy | 0.9745 |
|
| 59 |
+
| Loss | 0.0914 |
|
| 60 |
|
| 61 |
## Comparison with Sweep Models
|
| 62 |
|
|
|
|
| 96 |
- **Optimiser:** AdamW
|
| 97 |
- **Epochs:** 50 (sweep) + 5 (HPO final retrain)
|
| 98 |
- **FP16:** enabled
|
| 99 |
+
- **Dataset:** Kvasir-SEG augmented (4,800 train / 100 val / 100 test)
|
| 100 |
- **Augmentation:** random H/V flips, ±30° rotation, brightness/contrast/saturation ±20 %
|
| 101 |
|
| 102 |
## How to Use
|
config.json
CHANGED
|
@@ -18,12 +18,13 @@
|
|
| 18 |
"optimiser": "AdamW"
|
| 19 |
},
|
| 20 |
"test_metrics": {
|
| 21 |
-
"dice": 0.
|
| 22 |
-
"iou": 0.
|
| 23 |
-
"f1": 0.
|
| 24 |
-
"precision": 0.
|
| 25 |
-
"recall": 0.
|
| 26 |
-
"accuracy": 0.
|
|
|
|
| 27 |
},
|
| 28 |
"dataset": "andreribeiro87/kvasir-seg-augmented",
|
| 29 |
"hpo": "Optuna MedianPruner, 60 trials (28 completed, 32 pruned), best eval_loss=0.0537"
|
|
|
|
| 18 |
"optimiser": "AdamW"
|
| 19 |
},
|
| 20 |
"test_metrics": {
|
| 21 |
+
"dice": 0.9233709573745728,
|
| 22 |
+
"iou": 0.8576500415802002,
|
| 23 |
+
"f1": 0.9233709475438584,
|
| 24 |
+
"precision": 0.9474478438948487,
|
| 25 |
+
"recall": 0.9004874267475651,
|
| 26 |
+
"accuracy": 0.9744630432128907,
|
| 27 |
+
"loss": 0.09141451492905617
|
| 28 |
},
|
| 29 |
"dataset": "andreribeiro87/kvasir-seg-augmented",
|
| 30 |
"hpo": "Optuna MedianPruner, 60 trials (28 completed, 32 pruned), best eval_loss=0.0537"
|