Kvasir Unet Scratch
This is a custom U-Net built from scratch for polyp segmentation in endoscopic images.
Modellquelle
- Art
- Other
- Dateigröße
- 1,86 MB
- Version
- vRevision f3cd397
- Aktualisiert
- 27. Sept. 2026
- Downloads
- 12
- Aufgaben
- Bildsegmentierung
- Quelle
- Hugging Face
Quellenbeschreibung
Kvasir-SEG Custom U-Net
This is a custom U-Net built from scratch for polyp segmentation in endoscopic images.
Model Details
- Architecture: 4-layer U-Net with BatchNorm and Dropout2d.
- Loss Function: Focal + Dice Loss (So the Network sees the small polyps)
- Resolution: Trained on 256x256 images.
- Data Augmentations: Spatial augmentations and color variation
Performance
- Validation IoU: 0.463803
Quellen
1 Quelle
Quellen
1 QuelleVerifiziert 27. Sept.
Modellartefakte
1 Artefakt
Modellartefakte
1 ArtefaktQuellenauszüge
2 Auszüge
Quellenauszüge
2 Auszüge--- license: mit tags: - pytorch - image-segmentation - medical - kvasir-seg --- # Kvasir-SEG Custom U-Net This is a custom U-Net built from scratch for polyp segmentation in endoscopic images. ## Model Details * **Architecture:** 4-layer U-Net with BatchNorm and Dropout2d. * **Loss Function:** Focal + Dice Loss (So the Network sees the small polyps) * **Resolution:** Trained on 256x256 images. * **Data Augmentations:** Spatial augmentations and color variation ## Performance * **Validation IoU:** 0.463803
Source context: 12 downloads · 0 likes · Pipeline image-segmentation · Repo Shushunya/kvasir-unet-scratch
