Update README.md

README.md

@@ -68,7 +68,7 @@ Segmentation Mask (256×256×1)
 | Hyperparameter | Value |
 |---------------|-------|
 | Backbone | DINOv3-ViT-L/16 (frozen) |
-| Multi-scale Layers | [5, 10, 16, 18, 20, 22, 23] |
+| Multi-scale Layers | [5, 11, 17, 20, 23] |
 | Input Resolution | 256×256 |
 | Batch Size | 96 |
 | Epochs | 150 |
@@ -96,22 +96,32 @@ Segmentation Mask (256×256×1)
 
 | Metric | Score |
 |--------|-------|
-| **Dice Score** | **{test_dice:.4f} ± {test_dice_std:.4f}** |
-| **IoU** | **{test_iou:.4f} ± {test_iou_std:.4f}** |
-| **Precision** | {test_precision:.4f} ± {test_precision_std:.4f} |
-| **Recall** | {test_recall:.4f} ± {test_recall_std:.4f} |
-| **HD95 (pixels)** | {test_hd95:.2f} ± {test_hd95_std:.2f} |
-| **Best Validation Dice** | {best_dice:.4f} |
+| **Dice Score** | **0.8289 ± 0.0000** |
+| **IoU** | **0.7078 ± 0.0000** |
+| **Precision** | 0.7910 ± 0.0000 |
+| **Recall** | 0.8705 ± 0.0000 |
+| **HD95 (pixels)** | 45.46 ± 0.00 |
+| **Best Validation Dice** | 0.7327 |
 
 ### Validation Set Results
 
 | Metric | Score |
 |--------|-------|
-| **Dice Score** | {val_dice:.4f} ± {val_dice_std:.4f} |
-| **IoU** | {val_iou:.4f} ± {val_iou_std:.4f} |
-| **Precision** | {val_precision:.4f} ± {val_precision_std:.4f} |
-| **Recall** | {val_recall:.4f} ± {val_recall_std:.4f} |
-| **HD95 (pixels)** | {val_hd95:.2f} ± {val_hd95_std:.2f} |
+| **Dice Score** | **0.8795 ± 0.0304** |
+| **IoU** | **0.7862 ± 0.0485** |
+| **Precision** | 0.8846 ± 0.0256 |
+| **Recall** | 0.8744 ± 0.0351 |
+| **HD95 (pixels)** | 30.85 ± 8.90 |
+
+### Training Set Results (Final Epoch)
+
+| Metric | Score |
+|--------|-------|
+| **Dice Score** | 0.8747 ± 0.0108 |
+| **IoU** | 0.7775 ± 0.0170 |
+| **Precision** | 0.8698 ± 0.0189 |
+| **Recall** | 0.8801 ± 0.0136 |
+| **HD95 (pixels)** | 33.91 ± 1.69 |
 
 ## Usage
 
@@ -241,10 +251,10 @@ If you use this model in your research, please cite:
 
 bibtex
 @software{dinov3_polyp_seg,
-  author = {Your Name},
+  author = {Amirreza Mehrzadian},
   title = {DINOv3 Polyp Segmentation with U-Net Decoder},
   year = {2024},
-  url = {https://huggingface.co/your-username/dinov3-polyp-seg}
+  url = {https://huggingface.co/uncleMehrzad/dinov3-polyp-seg}
 }
 ## Acknowledgments
 DINOv3 team for the powerful vision backbone