Kvasir Unet Scratch
This is a custom U-Net built from scratch for polyp segmentation in endoscopic images.
Fonte do modelo
- Tipo
- Other
- Tamanho do arquivo
- 1,86 MB
- Versão
- vRevision f3cd397
- Atualizado
- 27 de set. de 2026
- Downloads
- 12
- Tarefas
- Segmentação de imagem
- Fonte
- Hugging Face
Descrição da fonte
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
Fontes
1 fonte
Fontes
1 fonteVerificado 27 de set.
Artefatos de modelo
1 artefato
Artefatos de modelo
1 artefatoTrechos de fonte
2 trechos
Trechos de fonte
2 trechos--- 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
