bigmb/Unet-Segmentation-Pytorch-Nest-of-Unets

Health v2 · maintenance only

Flags: no_releases

Adoption not part of the score

2249 stars · 365 forks observed · 2026-08-28

What it is AI-extracted, prompt v1, taxonomy v1, 2026-08-30, confidence not recorded

A PyTorch implementation of several U-Net variants for image segmentation, including UNet, R2U-Net, Attention U-Net, Attention R2U-Net, and Nested UNet (UNet++). It includes training scripts, Dice score evaluation, and layer visualization utilities.

Use cases

• segment medical images with unet models in pytorch
• compare unet, attention unet, and nested unet architectures
• train a segmentation model on my own image dataset
• implement unet++ for image segmentation
• compute dice score for segmentation predictions

When to choose

• you want ready-to-run PyTorch code for multiple U-Net variants in one place
• you are doing research or learning about segmentation architectures like Attention U-Net or UNet++
• you work on medical or biomedical image segmentation tasks

When to avoid

• you need a production-grade, pip-installable segmentation library with an API
• you require models beyond the U-Net family or modern architectures like transformers
• you need actively maintained code with recent updates and community support

Facets

library · maturity maintenance

machine-learning deep-learning image-processing computer-vision deep-learning computer-vision image-processing healthcare python cross-platform unet semantic-segmentation pytorch attention-unet nested-unet medical-imaging research-code gpu

Member repositories

For agents

markdown · JSON · MCP: product_card(name="bigmb/Unet-Segmentation-Pytorch-Nest-of-Unets")