andreribeiro87 commited on
Commit
6dc1810
·
verified ·
1 Parent(s): a78ad5d

Upload unet3plus+efficientnet model, code, and model card

Browse files
Files changed (2) hide show
  1. README.md +2 -1
  2. 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 (5,368 train / 53 val / 53 test)
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.9234,
22
- "iou": 0.8577,
23
- "f1": 0.9234,
24
- "precision": 0.9474,
25
- "recall": 0.9005,
26
- "accuracy": 0.9745
 
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"