Kvasir Unet Scratch
This is a custom U-Net built from scratch for polyp segmentation in endoscopic images.
Fuente del modelo
- Tipo
- Other
- Tamaño del archivo
- 1,86 MB
- Versión
- vRevision f3cd397
- Actualizado
- 27 sept 2026
- Descargas
- 12
- Tareas
- Segmentación de imágenes
- Fuente
- Hugging Face
Descripción de la fuente
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
Fuentes
1 fuente
Fuentes
1 fuenteVerificado 27 sept
Artefactos del modelo
1 artefacto
Artefactos del modelo
1 artefactoExtractos de fuentes
2 extractos
Extractos de fuentes
2 extractos--- 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
