Beta
Explore
Marketplace
Neural Labs
Chat
Wallet
Docs
kvasir-unet-scratch – AI Model by Shushunya | AlphaNeural AI
You can deploy this model and start earning money today!
Shushunya
/
kvasir-unet-scratch
like
0
safetensors
pytorch
image-segmentation
medical
kvasir-seg
mit
us
Views
No views yet
Model card
Files and Versions
Community
API
Deploy
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
